Searched refs:TrtWeightStore (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 231 friend class TrtWeightStore; variable
241 class TrtWeightStore {
363 TrtWeightStore* weight_store,
371 TrtWeightStore* weight_store);
378 TrtWeightStore* weight_store;
423 TrtWeightStore weight_store_;
647 TrtWeightStore weight_store_;
|
| D | convert_nodes_test.cc | 433 TrtWeightStore store; in TEST()
790 TrtWeightStore* weight_store_;
939 Converter* converter, TrtWeightStore* weight_store, in TestPrepareTensorForShape()
1052 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange()
|
| D | convert_nodes.cc | 1087 TRT_ShapedWeights TrtWeightStore::GetTempWeights(nvinfer1::DataType trt_dtype, in GetTempWeights()
1103 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store, in OpConverterParams()
1118 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store) in OpConverterParams()
4307 Status TfTensorToTrtWeights(const Tensor& tensor, TrtWeightStore* weight_store, in TfTensorToTrtWeights()
|