Home
last modified time | relevance | path

Searched defs:GetInputTensors (Results 1 – 25 of 46) 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
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_dataset_op_test.cc41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon4a2f49560111::RewriteDatasetParams
Doptimize_dataset_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon8eda82050111::OptimizeDatasetParams
Dget_options_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon6e228bba0111::GetOptionsParams
Dreduce_dataset_op_test.cc50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonac3e1e8c0111::ReduceDatasetParams
Dzip_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon682d11070111::ZipDatasetParams
Dskip_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon0bb28a0b0111::SkipDatasetParams
Dfilter_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anona5d8115a0111::FilterDatasetParams
Dprefetch_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon467506450111::PrefetchDatasetParams
Dtf_record_dataset_op_test.cc33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon42b8eb8c0111::TFRecordDatasetParams
Dtext_line_dataset_op_test.cc39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon2a2748a00111::TextLineDatasetParams
Dflat_map_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonf474da580111::FlatMapDatasetParams
Dmap_defun_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon8ac0845e0111::MapDefunOpParams
Dparallel_batch_dataset_op_test.cc49 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon45b721a20111::ParallelBatchDatasetParams
Dshard_dataset_op_test.cc40 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon0da156460111::ShardDatasetParams
Dfixed_length_record_dataset_op_test.cc47 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon20f34b230111::FixedLengthRecordDatasetParams
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.cc1020 std::vector<Tensor> RangeDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::RangeDatasetParams
1043 std::vector<Tensor> BatchDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::BatchDatasetParams
1066 std::vector<Tensor> MapDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::MapDatasetParams
1102 std::vector<Tensor> TensorSliceDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::TensorSliceDatasetParams
1149 std::vector<Tensor> TakeDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::TakeDatasetParams
1168 std::vector<Tensor> ConcatenateDatasetParams::GetInputTensors() const { in GetInputTensors() function in tensorflow::data::ConcatenateDatasetParams
1188 std::vector<Tensor> OptionsDatasetParams::GetInputTensors() const { return {}; } in GetInputTensors() function in tensorflow::data::OptionsDatasetParams
/external/tensorflow/tensorflow/core/common_runtime/
Dsimple_propagator_state.h121 Entry* GetInputTensors(const TaggedNode& tagged_node) { in GetInputTensors() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drandom_dataset_op_test.cc56 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon1373acb20111::RandomDatasetParams
Dlmdb_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon5019c0b00111::LMDBDatasetParams
Dsampling_dataset_op_test.cc41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anone4efb82c0111::SamplingDatasetParams
Dassert_next_dataset_op_test.cc42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon2a326c810111::AssertNextDatasetParams
Dunique_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon085fcb880111::UniqueDatasetParams
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dbenchmark_mlir_function.cc50 static llvm::SmallVector<Tensor> GetInputTensors( in GetInputTensors() function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api_test.cc214 std::vector<std::vector<Tensor>> GetInputTensors() { in GetInputTensors() function in tensorflow::tensorrt::TrtConverterTest

12