Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h114 inline bool HasStaticShape(const nvinfer1::Dims& dims) { in HasStaticShape() function
122 inline bool HasStaticShape(std::vector<int> dims) { in HasStaticShape() function
Dconvert_nodes.cc604 if (!HasStaticShape(dims)) return -1; in TrtTensorDimsNumElements()
620 if (!HasStaticShape(lhs) || !HasStaticShape(rhs)) return false; in AreDimsStaticWithSameSize()
626 if (!HasStaticShape(lhs) || !HasStaticShape(rhs)) return false; in AreDimsStaticWithDifferentSize()
1723 if (input.is_weights() && !HasStaticShape(dims)) { in PrepareTensorForShape()
2540 if (HasStaticShape(inputs.at(0).GetTrtDims())) { in ConvertShape()
2662 if (!params->use_implicit_batch && !HasStaticShape(input_dims)) { in ConvertExpandDims()
2764 if (!params->use_implicit_batch && !HasStaticShape(*input_dims)) { in SqueezeTensor()
2799 if (params->use_implicit_batch || !HasStaticShape(dims)) { in ConvertSqueeze()
4762 if (HasStaticShape(inputs.at(i).GetTrtDims())) { in ConvertPack()
4783 !HasStaticShape(dims)) { in ConvertPack()
[all …]
Dconvert_nodes_test.cc1546 if (!converter_->use_implicit_batch() || HasStaticShape(trt_dims)) { in AddTestTensorWithTFDims()
1567 if (HasStaticShape(dims)) { in AddTestTensor()