Searched refs:transpose_a_ (Results 1 – 4 of 4) sorted by relevance
43 …transpose_a_ = primitive->transpose_a() ? nvinfer1::MatrixOperation::kTRANSPOSE : nvinfer1::Matrix… in AddInnerOp()46 transpose_a_ = nvinfer1::MatrixOperation::kNONE; in AddInnerOp()66 …auto matmul_layer = network->addMatrixMultiply(*matmul_input, transpose_a_, *weight, transpose_b_); in AddInnerOp()
39 nvinfer1::MatrixOperation transpose_a_ = nvinfer1::MatrixOperation::kNONE;
80 transpose_a_ = transpose_a_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()411 if (transpose_a_) { in GenerateStrategies()518 if (transpose_a_) { in PrepareStrategy()538 if (transpose_a_) { in InitTensorInfoForCost()
62 bool transpose_a_ = false; variable