Searched refs:weight_store (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 363 TrtWeightStore* weight_store,
371 TrtWeightStore* weight_store);
378 TrtWeightStore* weight_store; member
|
| D | convert_nodes.cc | 496 params->weight_store->GetTempWeights(trt_type, dims);
1103 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store, in OpConverterParams() argument
1110 weight_store(weight_store), in OpConverterParams()
1118 std::vector<TRT_TensorOrWeights>* outputs, TrtWeightStore* weight_store) in OpConverterParams() argument
1124 weight_store(weight_store), in OpConverterParams()
2359 params->weight_store->GetTempWeights(weights_rsck); in ConvertConv2DHelper()
2545 TRT_ShapedWeights weight = params->weight_store->GetTempWeights( in ConvertShape()
3301 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()
3350 params->weight_store->GetTempWeights(weights_drsck); in ConvertConv3DHelper()
3634 params->weight_store->GetTempWeights(weights); in ConvertFusedConv2DBiasActivation()
[all …]
|
| D | convert_nodes_test.cc | 939 Converter* converter, TrtWeightStore* weight_store, in TestPrepareTensorForShape() argument
947 input = TRT_TensorOrWeights(weight_store->GetTempWeights( in TestPrepareTensorForShape()
1052 void TestGetWeightRange(ConverterTest* test, TrtWeightStore* weight_store) { in TestGetWeightRange() argument
1056 weight_store->GetTempWeights(trt_type, GetTestDims({2, 3})); in TestGetWeightRange()
|