Searched refs:tensor_r (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4481 nvinfer1::ITensor* tensor_r = nullptr; in ConvertBinary() local
4488 &tensor_r, node_def, /*op_instance=*/1)); in ConvertBinary()
4493 *tensor_l, *tensor_r, op_pair->second); in ConvertBinary()
5625 nvinfer1::ITensor* tensor_r = nullptr; in ConvertBatchMatMul() local
5631 params->validation_only, &tensor_r, node_def)); in ConvertBatchMatMul()
5635 TRT_TensorOrWeights(tensor_r), transpose_a, in ConvertBatchMatMul()
5883 nvinfer1::ITensor* tensor_r = nullptr; in ConvertSquaredDifference() local
5889 params->validation_only, &tensor_r, node_def)); in ConvertSquaredDifference()
5895 *tensor_l, *tensor_r, nvinfer1::ElementWiseOperation::kSUB); in ConvertSquaredDifference()
|