Searched refs:operand_r (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 709 const TRT_TensorOrWeights& operand_r,
|
| D | convert_nodes.cc | 361 const TRT_TensorOrWeights& operand_r, in GetTrtBroadcastShape() argument
389 if (!operand_l.is_tensor() && !operand_r.is_tensor()) { in GetTrtBroadcastShape()
434 operand_r.GetTrtDims().nbDims + in GetTrtBroadcastShape()
435 (use_implicit_batch && operand_r.is_tensor())); in GetTrtBroadcastShape()
439 TF_RETURN_IF_ERROR(compute_output_dims(operand_r, broadcast_num_dims, in GetTrtBroadcastShape()
4468 const TRT_TensorOrWeights& operand_r = inputs.at(1); in ConvertBinary() local
4478 operand_l, operand_r, /*check_feasibility=*/true, in ConvertBinary()
4487 params->converter, operand_r, broadcasted_dims_r, params->validation_only, in ConvertBinary()
|