Home
last modified time | relevance | path

Searched refs:GetTensorOrWeights (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc761 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output) { in GetTensorOrWeights() function in tensorflow::tensorrt::convert::ConverterTest
762 return converter_->GetTensorOrWeights(name, output); in GetTensorOrWeights()
817 TF_EXPECT_OK(GetTensorOrWeights("my_op", &actual_output_1)); in TEST_F()
822 TF_EXPECT_OK(GetTensorOrWeights("my_op:1", &actual_output_2)); in TEST_F()
1043 TF_EXPECT_OK(GetTensorOrWeights("my_tensor", &added_tensor)); in TEST_F()
1365 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output) { in GetTensorOrWeights() function in tensorflow::tensorrt::convert::OpConverterTest
1366 return converter_->GetTensorOrWeights(name, output); in GetTensorOrWeights()
1684 TF_EXPECT_OK(GetTensorOrWeights(name.c_str(), &output)); in RunValidationAndConversion()
2045 TF_EXPECT_OK(test->GetTensorOrWeights("my_const", &output)); in TestConvertConst()
2402 TF_EXPECT_OK(GetTensorOrWeights("my_reshape", &output)); in TEST_F()
[all …]
Dconvert_nodes.h624 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output);
Dconvert_nodes.cc1365 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()
1570 Status Converter::GetTensorOrWeights(const string& name, in GetTensorOrWeights() function in tensorflow::tensorrt::convert::Converter