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::__anon9beb2eb90111::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::__anon1f8898000111::OptimizeDatasetParams
Dskip_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anone206da860111::SkipDatasetParams
Dzip_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon6eabbe220111::ZipDatasetParams
Dreduce_dataset_op_test.cc50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon68e879c70111::ReduceDatasetParams
Dflat_map_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon8522f0530111::FlatMapDatasetParams
Dtext_line_dataset_op_test.cc39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anond0981dfb0111::TextLineDatasetParams
Dprefetch_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anond7231c400111::PrefetchDatasetParams
Dshard_dataset_op_test.cc40 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonae7fb6210111::ShardDatasetParams
Dfilter_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon62826c950111::FilterDatasetParams
Dtf_record_dataset_op_test.cc33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anone929c0e70111::TFRecordDatasetParams
Dfixed_length_record_dataset_op_test.cc47 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon1e26ce3e0111::FixedLengthRecordDatasetParams
Dparallel_batch_dataset_op_test.cc46 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon213a87dd0111::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::__anon077d7b360111::AutoShardDatasetParams
Dsampling_dataset_op_test.cc41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anonfd106e070111::SamplingDatasetParams
Dassert_next_dataset_op_test.cc42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon320b187c0111::AssertNextDatasetParams
Drandom_dataset_op_test.cc55 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon10a72fcd0111::RandomDatasetParams
Dunique_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon05934ea30111::UniqueDatasetParams
Dlmdb_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon58cbc10b0111::LMDBDatasetParams
Ddirected_interleave_dataset_op_test.cc48 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon57b9fbde0111::DirectedInterleaveDatasetParams

12