Searched refs:TensorShapeType (Results 1 – 6 of 6) sorted by relevance
127 template <typename TensorShapeType>128 inline bool IsTrtShapeTensorCompatible(const TensorShapeType& shape) { in IsTrtShapeTensorCompatible()140 template <typename TensorShapeType>141 inline nvinfer1::Dims TensorShapeToTrtDims(const TensorShapeType& shape, in TensorShapeToTrtDims()159 template <typename TensorShapeType>161 TensorShapeType* shape,
90 using TensorShapeType = std::pair<DataType, TensorShape>; variable93 TensorShapeType>>;158 static const TensorShapeType* GetTensorShapeType(161 static const TensorShapeType* GetTensorShapeType(
378 std::priority_queue<std::tuple<int, const TensorShapeType*>> queue; in AddOutputTensorShapeTypeByTensorShapeMap()388 const TensorShapeType* tst = std::get<1>(queue.top()); in AddOutputTensorShapeTypeByTensorShapeMap()521 /* static */ const RemoteFusedGraphExecuteUtils::TensorShapeType*532 /* static */ const RemoteFusedGraphExecuteUtils::TensorShapeType*
316 const RemoteFusedGraphExecuteUtils::TensorShapeType* tst = in TEST()362 const RemoteFusedGraphExecuteUtils::TensorShapeType* tst = in TEST()
29 template <typename TensorShapeType>30 std::vector<nvinfer1::Dims> GetDimVec(std::vector<TensorShapeType> shape_vec) { in GetDimVec()32 absl::c_transform(shape_vec, dimvec.begin(), [](TensorShapeType shape) { in GetDimVec()
31 template <typename TensorShapeType> // TensorShape or PartialTensorShape32 Status AddDenseOutputShapes(const std::vector<TensorShapeType>& dense_shapes, in AddDenseOutputShapes()