Searched refs:weight_store_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 423 TrtWeightStore weight_store_; variable
647 TrtWeightStore weight_store_; variable
|
| D | convert_nodes_test.cc | 744 weight_store_ = &converter_->weight_store_; in Reset()
790 TrtWeightStore* weight_store_; member in tensorflow::tensorrt::convert::ConverterTest
976 converter_.get(), weight_store_, in TEST_F()
982 converter_.get(), weight_store_); in TEST_F()
987 weight_store_); in TEST_F()
993 converter_.get(), weight_store_); in TEST_F()
1002 weight_store_); in TEST_F()
1008 weight_store_, error::INVALID_ARGUMENT, in TEST_F()
1068 TestGetWeightRange<float>(this, weight_store_); in TEST_F()
1069 TestGetWeightRange<Eigen::half>(this, weight_store_); in TEST_F()
[all …]
|
| D | convert_nodes.cc | 1223 &weight_store_, precision_mode_, use_calibration_, in IsTensorRTCandidate()
1233 OpConverterParams params(const_node_def, inputs, &outputs, &weight_store_, in ConvertConstToWeights()
1290 OpConverterParams params(this, node_def, inputs, &outputs, &weight_store_); in ConvertNode()
|