Searched refs:GetTrtBroadcastShape (Results 1 – 3 of 3) 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() function
4224 TF_RETURN_IF_ERROR(GetTrtBroadcastShape(inputs.at(0), inputs.at(1), in ConvertBiasAdd()
4477 TF_RETURN_IF_ERROR(GetTrtBroadcastShape( in ConvertBinary()
5621 TF_RETURN_IF_ERROR(GetTrtBroadcastShape( in ConvertBatchMatMul()
5879 TF_RETURN_IF_ERROR(GetTrtBroadcastShape( in ConvertSquaredDifference()
|
| D | convert_nodes_test.cc | 1144 TEST_F(ConverterTest, GetTrtBroadcastShape) { in TEST_F() argument
1176 GetTrtBroadcastShape(operand_1, operand_2, /*check_feasibility=*/true, in TEST_F()
1186 GetTrtBroadcastShape(operand_2, operand_1, /*check_feasibility=*/true, in TEST_F()
|