Home
last modified time | relevance | path

Searched refs:GetInputs (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dinput_generator_test.cc48 ASSERT_EQ(input_generator.GetInputs(), inputs); in TEST()
65 auto inputs = input_generator.GetInputs(); in TEST()
75 auto inputs = input_generator.GetInputs(); in TEST()
Dutil.h83 auto input_tensor_ids = runner->GetInputs(); in RunKernelTest()
84 auto inputs = input_generator.GetInputs(); in RunKernelTest()
Dinput_generator.h38 std::vector<string> GetInputs();
Dinput_generator.cc205 std::vector<string> InputGenerator::GetInputs() { return inputs_; } in GetInputs() function in tflite::testing::InputGenerator
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver_test.cc33 ASSERT_THAT(runner->GetInputs(), ElementsAre(0, 1, 2, 3)); in TEST()
69 ASSERT_THAT(runner->GetInputs(), ElementsAre(0, 1, 2, 3)); in TEST()
Dtflite_driver.h39 const std::vector<int>& GetInputs() override { in GetInputs() function
Dtest_runner.h39 virtual const std::vector<int>& GetInputs() = 0;
Dtf_driver.h46 const std::vector<int>& GetInputs() override { return input_ids_; } in GetInputs() function
Dtest_runner_test.cc27 const std::vector<int>& GetInputs() override { return ids_; } in GetInputs() function in tflite::testing::__anon73a85d540111::ConcreteTestRunner
Dtf_driver_test.cc99 ASSERT_THAT(runner->GetInputs(), ElementsAre(0, 1, 2, 3)); in TEST()
Dparse_testdata.cc258 expected_inputs_ = test_runner->GetInputs(); in Invoke()
299 expected_inputs_ = test_runner->GetInputs(); in Reshape()
/external/tensorflow/tensorflow/cc/framework/
Dops.cc21 Operation::Operation(Node* n) : inputs_(GetInputs(n)), node_(n) {} in Operation()
52 Operation::Inputs Operation::GetInputs(Node* node) { in GetInputs() function in tensorflow::Operation
Dops.h58 static Inputs GetInputs(Node* node);
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc108 ThreeFry2x32State GetInputs(const int64 size, XlaBuilder* builder) { in GetInputs() function
120 ThreeFry2x32State inputs = GetInputs(half_size, builder); in StatelessRngUniformU32()
147 ThreeFry2x32State inputs = GetInputs(size, builder); in StatelessRngUniformU64()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.h41 const std::vector<int>& GetInputs() override { in GetInputs() function
Dtest_runner.h42 virtual const std::vector<int>& GetInputs() = 0;
Dparse_testdata.cc261 expected_inputs_ = test_runner->GetInputs(); in Invoke()
302 expected_inputs_ = test_runner->GetInputs(); in Reshape()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h529 Status GetInputs(const NodeDef& node_def,
Dconvert_nodes_test.cc547 Status GetInputs(const NodeDef& node_def, in GetInputs() function in tensorflow::tensorrt::convert::ConverterTest
549 return converter_->GetInputs(node_def, inputs); in GetInputs()
629 TF_EXPECT_OK(GetInputs(node_def, &inputs)); in TEST_F()
Dconvert_nodes.cc1003 TF_RETURN_IF_ERROR(this->GetInputs(node_def, &inputs)); in ConvertNode()
1381 Status Converter::GetInputs(const NodeDef& node_def, in GetInputs() function in tensorflow::tensorrt::convert::Converter
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc126 Status GetInputs(NodeMap* node_map, const std::vector<NodeDef*>& ops, in GetInputs() function
289 GetInputs(sa_opti->node_map(), ops, *dtype, inputs)); in AnalyzeInputs()