Home
last modified time | relevance | path

Searched refs:GetInputTensors (Results 1 – 25 of 40) sorted by relevance

12

/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dbase_task_api.h87 std::vector<TfLiteTensor*> GetInputTensors() { return engine_->GetInputs(); } in GetInputTensors() function
101 RETURN_IF_ERROR(Preprocess(GetInputTensors(), args...)); in Infer()
121 RETURN_IF_ERROR(Preprocess(GetInputTensors(), args...)); in InferWithFallback()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonfe0e44000111::MapDefunOpParams
244 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_P()
273 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_F()
Ddataset_test_base.h130 virtual std::vector<Tensor> GetInputTensors() const = 0;
189 std::vector<Tensor> GetInputTensors() const override;
225 std::vector<Tensor> GetInputTensors() const override;
265 std::vector<Tensor> GetInputTensors() const override;
290 std::vector<Tensor> GetInputTensors() const override;
330 std::vector<Tensor> GetInputTensors() const override;
362 std::vector<Tensor> GetInputTensors() const override;
Ddataset_test_base.cc828 for (auto& input : dataset_params.GetInputTensors()) { in RunDatasetOp()
930 auto input_tensors = dataset_params.GetInputTensors(); in MakeDatasetTensor()
986 std::vector<Tensor> RangeDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::RangeDatasetParams
1010 std::vector<Tensor> BatchDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::BatchDatasetParams
1035 std::vector<Tensor> MapDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::MapDatasetParams
1073 std::vector<Tensor> TensorSliceDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::TensorSliceDatasetParams
1120 std::vector<Tensor> TakeDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::TakeDatasetParams
1140 std::vector<Tensor> ConcatenateDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::ConcatenateDatasetParams
Doptimize_dataset_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon490417670111::OptimizeDatasetParams
Dreduce_dataset_op_test.cc50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon3adef36e0111::ReduceDatasetParams
Dskip_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon40f4be6d0111::SkipDatasetParams
Dzip_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anone5e941290111::ZipDatasetParams
Dflat_map_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonae9e6fba0111::FlatMapDatasetParams
Dprefetch_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon009e9ba70111::PrefetchDatasetParams
Dtext_line_dataset_op_test.cc39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon29838a420111::TextLineDatasetParams
Dfilter_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon3478e63c0111::FilterDatasetParams
Dtf_record_dataset_op_test.cc33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon42152d2e0111::TFRecordDatasetParams
Dshard_dataset_op_test.cc40 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anoneb2a16e80111::ShardDatasetParams
Dfixed_length_record_dataset_op_test.cc47 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonf1ff7f450111::FixedLengthRecordDatasetParams
Dparallel_batch_dataset_op_test.cc46 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon57c7b8840111::ParallelBatchDatasetParams
/external/tensorflow/tensorflow/core/common_runtime/
Dsimple_propagator_state.h120 Entry* GetInputTensors(const TaggedNode& tagged_node) { in GetInputTensors() function
/external/tensorflow/tensorflow/lite/tools/optimize/
Dmodify_model_interface.cc48 std::vector<TensorOpTensor> GetInputTensors(const TensorType& input_type, in GetInputTensors() function
305 GetInputTensors(input_type, model, error_reporter); in GetOriginalNumberOfTensors()
337 GetInputTensors(input_type, model, &error_reporter); in ModifyModelInterface()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dauto_shard_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon8ac557bd0111::AutoShardDatasetParams
Dsampling_dataset_op_test.cc41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon29d97cce0111::SamplingDatasetParams
Dassert_next_dataset_op_test.cc42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon1e4e85e30111::AssertNextDatasetParams
Drandom_dataset_op_test.cc55 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anone47fe0d40111::RandomDatasetParams
Dunique_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anond96bffaa0111::UniqueDatasetParams
Dlmdb_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon3ae684520111::LMDBDatasetParams
Ddirected_interleave_dataset_op_test.cc48 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anonb12be0450111::DirectedInterleaveDatasetParams

12