Searched refs:transpose_b_ (Results 1 – 4 of 4) sorted by relevance
44 …transpose_b_ = primitive->transpose_b() ? nvinfer1::MatrixOperation::kTRANSPOSE : nvinfer1::Matrix… in AddInnerOp()47 transpose_b_ = nvinfer1::MatrixOperation::kTRANSPOSE; in AddInnerOp()66 …auto matmul_layer = network->addMatrixMultiply(*matmul_input, transpose_a_, *weight, transpose_b_); in AddInnerOp()
40 nvinfer1::MatrixOperation transpose_b_ = nvinfer1::MatrixOperation::kNONE; variable
91 transpose_b_ = transpose_b_iter->second->cast<BoolImmPtr>()->value(); in GetAttrs()172 if (!transpose_b_ && (mat_a_strategy.back() != mat_b_strategy.at(SECOND_FROM_END(mat_b_size)))) { in CheckStrategy()177 } else if (transpose_b_ && (mat_a_strategy.back() != mat_b_strategy.back())) { in CheckStrategy()211 SetDevMatrixShape(mat_a_strategy, mat_b_strategy, transpose_b_, &dev_matrix_shape_); in InferDevMatrixShape()283 if (transpose_b_) { in InferTensorMap()416 if (transpose_b_) { in GenerateStrategies()523 if (transpose_b_) { in PrepareStrategy()553 if (transpose_b_) { in InitTensorInfoForCost()
63 bool transpose_b_ = false; variable