| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| H A D | benchmark_mlir_function.cc | 50 static llvm::SmallVector<Tensor> GetInputTensors( in GetInputTensors() function
105 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunJitRtBenchmark()
179 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunTfrtBenchmark()
205 llvm::SmallVector<Tensor> input_tensors = GetInputTensors(input_specs); in RunEigenBenchmark()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| H A D | base_task_api.h | 87 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.h | 138 virtual std::vector<Tensor> GetInputTensors() const = 0;
206 std::vector<Tensor> GetInputTensors() const override;
242 std::vector<Tensor> GetInputTensors() const override;
282 std::vector<Tensor> GetInputTensors() const override;
308 std::vector<Tensor> GetInputTensors() const override;
349 std::vector<Tensor> GetInputTensors() const override;
381 std::vector<Tensor> GetInputTensors() const override;
405 std::vector<Tensor> GetInputTensors() const override;
|
| H A D | dataset_test_base.cc | 843 for (auto& input : dataset_params.GetInputTensors()) { in RunDatasetOp()
957 auto input_tensors = dataset_params.GetInputTensors(); in MakeDatasetTensor()
1020 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | map_defun_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon15242baf0111::MapDefunOpParams
245 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_P()
274 auto input_tensors = test_case.map_defun_op_params.GetInputTensors(); in TEST_F()
|
| H A D | rewrite_dataset_op_test.cc | 41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6ba169670111::RewriteDatasetParams
|
| H A D | optimize_dataset_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonde90a4360111::OptimizeDatasetParams
|
| H A D | get_options_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon200d4b4b0111::GetOptionsParams
|
| H A D | zip_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon1a17d0980111::ZipDatasetParams
|
| H A D | reduce_dataset_op_test.cc | 50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon76f4007d0111::ReduceDatasetParams
|
| H A D | skip_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonfaf53bbc0111::SkipDatasetParams
|
| H A D | flat_map_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon442afc890111::FlatMapDatasetParams
|
| H A D | prefetch_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon962b28760111::PrefetchDatasetParams
|
| H A D | text_line_dataset_op_test.cc | 39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon70a1b0f10111::TextLineDatasetParams
|
| H A D | tf_record_dataset_op_test.cc | 33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon893353dd0111::TFRecordDatasetParams
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | modify_model_interface.cc | 51 std::vector<TensorOpTensor> GetInputTensors(const TensorType& input_type, in GetInputTensors() function
308 GetInputTensors(input_type, model, error_reporter); in GetOriginalNumberOfTensors()
340 GetInputTensors(input_type, model, &error_reporter); in ModifyModelInterface()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| H A D | trt_convert_api_test.cc | 46 input_tensors_ = GetInputTensors(); in TrtConverterTest()
214 std::vector<std::vector<Tensor>> GetInputTensors() { in GetInputTensors() function in tensorflow::tensorrt::TrtConverterTest
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | simple_propagator_state.h | 121 Entry* GetInputTensors(const TaggedNode& tagged_node) { in GetInputTensors() function
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/ |
| H A D | OpWrapper.h | 95 const std::vector<std::shared_ptr<TensorWrapper>>& GetInputTensors() { in GetInputTensors() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | sampling_dataset_op_test.cc | 41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon6b5821fd0111::SamplingDatasetParams
|
| H A D | assert_next_dataset_op_test.cc | 42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon31a0d0b20111::AssertNextDatasetParams
|
| H A D | random_dataset_op_test.cc | 56 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon16656e430111::RandomDatasetParams
|
| H A D | unique_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon0b518d190111::UniqueDatasetParams
|
| H A D | lmdb_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anond9dbea010111::LMDBDatasetParams
|
| H A D | assert_prev_dataset_op_test.cc | 64 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon29239d900111::AssertPrevDatasetParams
|