Searched refs:tensor_r (Results 1 – 1 of 1) sorted by relevance
1973 const nvinfer1::ITensor* tensor_r = nullptr; in BinaryTensorOpTensor() local1978 operand_r, broadcasted_dims_r, /*validation_only=*/false, &tensor_r); in BinaryTensorOpTensor()1988 TFTRT_CHECK_EQ_TYPE(tensor_r->getType(), dtype) in BinaryTensorOpTensor()1989 << DebugString(tensor_r->getType()) << " vs " << DebugString(dtype); in BinaryTensorOpTensor()1995 *const_cast<nvinfer1::ITensor*>(tensor_r), op_pair->second); in BinaryTensorOpTensor()3907 const nvinfer1::ITensor* tensor_r; in ConvertBatchMatMul() local3909 TF_RETURN_IF_ERROR(get_tensor_with_proper_dims(inputs.at(1), &tensor_r)); in ConvertBatchMatMul()3915 *const_cast<nvinfer1::ITensor*>(tensor_r), transpose_b); in ConvertBatchMatMul()