Searched refs:operand_l (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 507 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l,
|
| 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()
424 std::max(operand_l.GetTrtDims().nbDims + (operand_l.is_tensor() ? 1 : 0), in GetTrtBroadcastShape()
427 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()
1936 const TRT_TensorOrWeights& operand_l, in BinaryTensorOpTensor() argument
1957 operand_l, operand_r, &broadcasted_dims_l, &broadcasted_dims_r); in BinaryTensorOpTensor()
1975 operand_l, broadcasted_dims_l, /*validation_only=*/false, &tensor_l); in BinaryTensorOpTensor()
|