| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| 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()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon9257d62f0111::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()
|
| D | rewrite_dataset_op_test.cc | 41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6c8af3e70111::RewriteDatasetParams
|
| D | optimize_dataset_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonfcab7eb60111::OptimizeDatasetParams
|
| D | get_options_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anonb8d595cb0111::GetOptionsParams
|
| D | zip_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anonb2e01b180111::ZipDatasetParams
|
| D | reduce_dataset_op_test.cc | 50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anone3963afd0111::ReduceDatasetParams
|
| D | skip_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonacc6d63c0111::SkipDatasetParams
|
| D | flat_map_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6245d7090111::FlatMapDatasetParams
|
| D | prefetch_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonb44602f60111::PrefetchDatasetParams
|
| D | text_line_dataset_op_test.cc | 39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon5217db710111::TextLineDatasetParams
|
| D | tf_record_dataset_op_test.cc | 33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon6aa97e5d0111::TFRecordDatasetParams
|
| D | filter_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anondd302dcb0111::FilterDatasetParams
|
| /external/tensorflow/tensorflow/core/data/ |
| 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;
|
| 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
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| 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
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | simple_propagator_state.h | 121 Entry* GetInputTensors(const TaggedNode& tagged_node) { in GetInputTensors() function
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| 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()
|
| /external/executorch/backends/qualcomm/aot/wrappers/ |
| D | OpWrapper.h | 95 const std::vector<std::shared_ptr<TensorWrapper>>& GetInputTensors() { in GetInputTensors() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | sampling_dataset_op_test.cc | 41 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon74e20c7d0111::SamplingDatasetParams
|
| D | assert_next_dataset_op_test.cc | 42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon3b20ab320111::AssertNextDatasetParams
|
| D | random_dataset_op_test.cc | 56 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anona772b8c30111::RandomDatasetParams
|
| D | unique_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon9c5ed7990111::UniqueDatasetParams
|
| D | lmdb_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon7a1494810111::LMDBDatasetParams
|