Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5503 const auto get_matrix_op = [](nvinfer1::ITensor* in, in ConvertMatMulHelper() local
5537 *tensor_a, get_matrix_op(tensor_a, transpose_a), *tensor_b, in ConvertMatMulHelper()
5538 get_matrix_op(tensor_b, transpose_b)); in ConvertMatMulHelper()