Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dinput_generator_test.cc47 ASSERT_EQ(input_generator.GetInputs(), inputs); in TEST()
64 auto inputs = input_generator.GetInputs(); in TEST()
74 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/core/common_runtime/eager/
Dcustom_device_op_handler.cc69 auto inputs = op->GetInputs(); in Execute()
114 if (!op.GetInputs().empty()) { in MaybePinToCustomDevice()
115 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
138 for (const ImmediateExecutionTensorHandle* generic_input : op.GetInputs()) { in MaybePinToCustomDevice()
Dcore.cc214 &pin_to_cpu, Name(), GetInputs(), ctx_.HostCPU()->name())); in Execute()
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver_test.cc33 ASSERT_THAT(runner->GetInputs(), ElementsAre(0, 1, 2, 3)); in TEST()
70 ASSERT_THAT(runner->GetInputs(), ElementsAre(0, 1, 2, 3)); in TEST()
105 ASSERT_THAT(runner->GetInputs(), ElementsAre(1)); in TEST()
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::__anonc1c12f320111::ConcreteTestRunner
Dtest_runner.h40 virtual const std::vector<int>& GetInputs() = 0;
Dtflite_driver.h58 const std::vector<int>& GetInputs() override { in GetInputs() function
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()
311 expected_inputs_ = test_runner->GetInputs(); in Reshape()
Dgenerate_testspec.cc115 auto inputs = runner->GetInputs(); in GenerateTestSpecFromRunner()
/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
/external/tensorflow/tensorflow/go/
Dsignature.go95 for name, input := range pb.GetInputs() {
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_operation.h45 virtual absl::Span<ImmediateExecutionTensorHandle* const> GetInputs()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc82 for (auto input : orig_op->GetInputs()) { in Run()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_runner.h42 virtual const std::vector<int>& GetInputs() = 0;
Dtflite_driver.h41 const std::vector<int>& GetInputs() override { in GetInputs() function
Dparse_testdata.cc261 expected_inputs_ = test_runner->GetInputs(); in Invoke()
302 expected_inputs_ = test_runner->GetInputs(); in Reshape()
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dbase_task_api.h87 std::vector<TfLiteTensor*> GetInputTensors() { return engine_->GetInputs(); } in GetInputTensors()
Dtflite_engine.h130 std::vector<TfLiteTensor*> GetInputs();
Dtflite_engine.cc77 std::vector<TfLiteTensor*> TfLiteEngine::GetInputs() { in GetInputs() function in tflite::task::core::TfLiteEngine
/external/libchrome/build/android/gyp/util/
Dproguard_util.py148 def GetInputs(self): member in ProguardCmdBuilder

12