Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h155 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims,
160 Status TrtDimsToTensorShape(const nvinfer1::Dims trt_dims,
Dutils.cc191 TF_RETURN_IF_ERROR(TrtDimsToTensorShape(input_dim, &input_shapes->at(i))); in GetNetworkInputShapes()
195 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims, in TrtDimsToTensorShape() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_engine_utils.cc99 TF_RETURN_IF_ERROR(TrtDimsToTensorShape( in GetTrtBindingShape()