Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5513 const auto prepare_matmul_operand = in ConvertMatMulHelper() local
5532 nvinfer1::ITensor* tensor_a = prepare_matmul_operand(input_a, &transpose_a); in ConvertMatMulHelper()
5533 nvinfer1::ITensor* tensor_b = prepare_matmul_operand(input_b, &transpose_b); in ConvertMatMulHelper()