Searched refs:AddTensorOrWeights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 523 Status AddTensorOrWeights(const string& name, TRT_TensorOrWeights input);
|
| D | convert_nodes_test.cc | 539 Status AddTensorOrWeights(const string& name, TRT_TensorOrWeights input) { in AddTensorOrWeights() function in tensorflow::tensorrt::convert::ConverterTest
540 return converter_->AddTensorOrWeights(name, input); in AddTensorOrWeights()
790 TF_EXPECT_OK(AddTensorOrWeights("my_tensor", tensor)); in TEST_F()
798 ExpectStatus(AddTensorOrWeights("my_tensor", tensor), error::ALREADY_EXISTS, in TEST_F()
1236 converter_->AddTensorOrWeights(name, TRT_TensorOrWeights{weights})); in AddTestWeights()
|
| D | convert_nodes.cc | 1041 Status status = AddTensorOrWeights(output_name, output); in ConvertNode()
1066 status = AddTensorOrWeights(name, TRT_TensorOrWeights(tensor)); in AddInputTensor()
1132 Status Converter::AddTensorOrWeights(const string& name, in AddTensorOrWeights() function in tensorflow::tensorrt::convert::Converter
|