Searched refs:AddTensorOrWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 621 Status AddTensorOrWeights(const string& name, TRT_TensorOrWeights input);
|
| D | convert_nodes_test.cc | 757 Status AddTensorOrWeights(const string& name, TRT_TensorOrWeights input) { in AddTensorOrWeights() function in tensorflow::tensorrt::convert::ConverterTest
758 return converter_->AddTensorOrWeights(name, input); in AddTensorOrWeights()
1039 TF_EXPECT_OK(AddTensorOrWeights("my_tensor", tensor)); in TEST_F()
1047 ExpectStatus(AddTensorOrWeights("my_tensor", tensor), error::ALREADY_EXISTS, in TEST_F()
1597 converter_->AddTensorOrWeights(name, TRT_TensorOrWeights{weights})); in AddTestWeights()
|
| D | convert_nodes.cc | 1322 Status status = AddTensorOrWeights(output_name, output); in ConvertNode()
1351 status = AddTensorOrWeights(name, TRT_TensorOrWeights(tensor)); in AddInputTensor()
1556 Status Converter::AddTensorOrWeights(const string& name, in AddTensorOrWeights() function in tensorflow::tensorrt::convert::Converter
|