Searched refs:operand_l (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 708 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l,
|
| D | convert_nodes.cc | 360 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l, in GetTrtBroadcastShape() argument
389 if (!operand_l.is_tensor() && !operand_r.is_tensor()) { in GetTrtBroadcastShape()
432 std::max(operand_l.GetTrtDims().nbDims + in GetTrtBroadcastShape()
433 (use_implicit_batch && operand_l.is_tensor()), in GetTrtBroadcastShape()
437 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()
4467 const TRT_TensorOrWeights& operand_l = inputs.at(0); in ConvertBinary() local
4478 operand_l, operand_r, /*check_feasibility=*/true, in ConvertBinary()
4484 params->converter, operand_l, broadcasted_dims_l, params->validation_only, in ConvertBinary()
|