Searched refs:operand_r (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 508 const TRT_TensorOrWeights& operand_r,
|
| D | convert_nodes.cc | 363 const TRT_TensorOrWeights& operand_l, const TRT_TensorOrWeights& operand_r, in GetTrtBroadcastShape() argument
390 if (!operand_l.is_tensor() && !operand_r.is_tensor()) { in GetTrtBroadcastShape()
425 operand_r.GetTrtDims().nbDims + (operand_r.is_tensor() ? 1 : 0)); in GetTrtBroadcastShape()
429 TF_RETURN_IF_ERROR(compute_output_dims(operand_r, broadcast_num_dims, in GetTrtBroadcastShape()
1937 const TRT_TensorOrWeights& operand_r) { in BinaryTensorOpTensor() argument
1957 operand_l, operand_r, &broadcasted_dims_l, &broadcasted_dims_r); in BinaryTensorOpTensor()
1978 operand_r, broadcasted_dims_r, /*validation_only=*/false, &tensor_r); in BinaryTensorOpTensor()
|