Searched refs:weight_store_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 413 TrtWeightStore weight_store_; variable
553 TrtWeightStore weight_store_; variable
|
| D | convert_nodes_test.cc | 526 weight_store_ = &converter_->weight_store_; in ConverterTest()
576 TrtWeightStore* weight_store_; member in tensorflow::tensorrt::convert::ConverterTest
751 weight_store_->GetTempWeights(DT_FLOAT, GetTestDims({2, 3, 5})); in TEST_F()
818 TestGetWeightRange<float>(this, weight_store_); in TEST_F()
819 TestGetWeightRange<Eigen::half>(this, weight_store_); in TEST_F()
820 TestGetWeightRange<int32>(this, weight_store_); in TEST_F()
977 weight_store_->GetTempWeights(dtype, GetTestDims({2, 3, 5})); in TEST_F()
1224 weights = converter_->weight_store_.GetTempWeights(dtype, trt_dims); in AddTestWeights()
|
| D | convert_nodes.cc | 976 /*arg_validation_only=*/true, &weight_store_); in ValidateNode()
987 /*arg_validation_only=*/true, &weight_store_); in ConvertConstToWeights()
1006 /*arg_validation_only=*/false, &weight_store_); in ConvertNode()
|