Home
last modified time | relevance | path

Searched refs:GetTrtBroadcastShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h507 Status GetTrtBroadcastShape(const TRT_TensorOrWeights& operand_l,
Dconvert_nodes_test.cc884 TEST_F(ConverterTest, GetTrtBroadcastShape) { in TEST_F() argument
916 this->converter_->GetTrtBroadcastShape( in TEST_F()
925 this->converter_->GetTrtBroadcastShape( in TEST_F()
Dconvert_nodes.cc362 Status Converter::GetTrtBroadcastShape( in GetTrtBroadcastShape() function in tensorflow::tensorrt::convert::Converter
1956 Status status = params->converter->GetTrtBroadcastShape( in BinaryTensorOpTensor()