Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1972 const nvinfer1::ITensor* tensor_l = nullptr; in BinaryTensorOpTensor() local
1975 operand_l, broadcasted_dims_l, /*validation_only=*/false, &tensor_l); in BinaryTensorOpTensor()
1986 TFTRT_CHECK_EQ_TYPE(tensor_l->getType(), dtype) in BinaryTensorOpTensor()
1987 << DebugString(tensor_l->getType()) << " vs " << DebugString(dtype); in BinaryTensorOpTensor()
1994 *const_cast<nvinfer1::ITensor*>(tensor_l), in BinaryTensorOpTensor()
3906 const nvinfer1::ITensor* tensor_l; in ConvertBatchMatMul() local
3908 TF_RETURN_IF_ERROR(get_tensor_with_proper_dims(inputs.at(0), &tensor_l)); in ConvertBatchMatMul()
3914 *const_cast<nvinfer1::ITensor*>(tensor_l), transpose_a, in ConvertBatchMatMul()