Home
last modified time | relevance | path

Searched refs:input2Shape (Results 1 – 1 of 1) sorted by relevance

/dports/misc/nn-insight/nn-insight-1.0.5/
H A Dcompute.cpp345 const float *input2, const TensorShape &input2Shape, in compute()
354 …if (input1Shape==input2Shape) { // 2 streams of the same size produce another stream of the same s… in compute()
359 …} else if (input2Shape.size()==1 && input2Shape[0]==1) { // operation with a constant from the mod… in compute()
364 } else if (Tensor::isSubset(input1Shape, input2Shape)) { // operation with a smaller vector in compute()
368 auto input2e = input2+Tensor::flatSize(input2Shape); in compute()
927 auto input2Shape = model->getTensorShape(inputs[1]); in compute() local
949 …<< operatorKind << " isn't yet implemented for shapes " << input1Shape << " and " << input2Shape)); in compute()
1137 auto input2Shape = model->getTensorShape(inputs[1]); in compute() local
1152 …<< operatorKind << " isn't yet implemented for shapes " << input1Shape << " and " << input2Shape)); in compute()