Home
last modified time | relevance | path

Searched defs:GetInputTensors (Results 1 – 25 of 47) 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::__anon6c8af3e70111::RewriteDatasetParams
Doptimize_dataset_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonfcab7eb60111::OptimizeDatasetParams
Dget_options_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anonb8d595cb0111::GetOptionsParams
Dreduce_dataset_op_test.cc50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anone3963afd0111::ReduceDatasetParams
Dzip_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anonb2e01b180111::ZipDatasetParams
Dskip_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonacc6d63c0111::SkipDatasetParams
Dfilter_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anondd302dcb0111::FilterDatasetParams
Dmap_defun_op_test.cc43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon9257d62f0111::MapDefunOpParams
Dtf_record_dataset_op_test.cc33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6aa97e5d0111::TFRecordDatasetParams
Dflat_map_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6245d7090111::FlatMapDatasetParams
Dprefetch_dataset_op_test.cc45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonb44602f60111::PrefetchDatasetParams
Dtext_line_dataset_op_test.cc39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon5217db710111::TextLineDatasetParams
Dparallel_batch_dataset_op_test.cc49 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon5c591f130111::ParallelBatchDatasetParams
Dshard_dataset_op_test.cc40 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anond13f28970111::ShardDatasetParams
/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/executorch/backends/qualcomm/aot/wrappers/
DOpWrapper.h95 const std::vector<std::shared_ptr<TensorWrapper>>& GetInputTensors() { in GetInputTensors() function
/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::__anona772b8c30111::RandomDatasetParams
Dlmdb_dataset_op_test.cc44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon7a1494810111::LMDBDatasetParams
Dsampling_dataset_op_test.cc41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon74e20c7d0111::SamplingDatasetParams
Dassert_next_dataset_op_test.cc42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon3b20ab320111::AssertNextDatasetParams
Dunique_dataset_op_test.cc38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon9c5ed7990111::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