Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.cc141 Status GetNetworkInputShapes(const nvinfer1::INetworkDefinition* network, in GetNetworkInputShapes() function
Dutils.h360 Status GetNetworkInputShapes(const nvinfer1::INetworkDefinition* network,
Dconvert_nodes_test.cc1297 GetNetworkInputShapes(converter_->network(), &input_partial_shapes)); in BuildAndRun()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_shape_optimization_profiles_test.cc210 TF_CHECK_OK(GetNetworkInputShapes(network_.get(), &input_partial_shapes)); in TEST_P()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_resource_ops_test.cc201 TF_CHECK_OK(GetNetworkInputShapes(network.get(), &input_partial_shapes)); in CreateTRTEngine()