Home
last modified time | relevance | path

Searched refs:GetNetworkInputShapes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h152 Status GetNetworkInputShapes(const nvinfer1::INetworkDefinition* network,
Dutils.cc184 Status GetNetworkInputShapes(const nvinfer1::INetworkDefinition* network, in GetNetworkInputShapes() function
Dconvert_nodes_test.cc1487 GetNetworkInputShapes(converter_->network(), &input_partial_shapes)); in BuildAndRun()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles_test.cc174 TF_CHECK_OK(GetNetworkInputShapes(network_.get(), &input_partial_shapes)); in TEST_F()