Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h166 int64_t TrtWeightDimsNumElements(const nvinfer1::Dims& dims);
Dconvert_nodes.cc594 int64_t TrtWeightDimsNumElements(const nvinfer1::Dims& dims) { in TrtWeightDimsNumElements() function
613 return TrtWeightDimsNumElements(lhs) == TrtWeightDimsNumElements(rhs); in DimsHaveSameSize()
702 return TrtWeightDimsNumElements(shape_); in count()
Dconvert_nodes_test.cc1586 const int64_t num_elements = TrtWeightDimsNumElements(trt_dims); in AddTestWeights()