| /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::__anon8ac0845e0111::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::__anon4a2f49560111::RewriteDatasetParams
|
| D | optimize_dataset_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon8eda82050111::OptimizeDatasetParams
|
| D | get_options_op_test.cc | 43 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon6e228bba0111::GetOptionsParams
|
| D | zip_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::__anon682d11070111::ZipDatasetParams
|
| D | skip_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon0bb28a0b0111::SkipDatasetParams
|
| D | reduce_dataset_op_test.cc | 50 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonac3e1e8c0111::ReduceDatasetParams
|
| D | flat_map_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anonf474da580111::FlatMapDatasetParams
|
| D | prefetch_dataset_op_test.cc | 45 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon467506450111::PrefetchDatasetParams
|
| D | text_line_dataset_op_test.cc | 39 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon2a2748a00111::TextLineDatasetParams
|
| D | shard_dataset_op_test.cc | 40 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon0da156460111::ShardDatasetParams
|
| D | tf_record_dataset_op_test.cc | 33 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::__anon42b8eb8c0111::TFRecordDatasetParams
|
| /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/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/tensorflow/tensorflow/core/common_runtime/ |
| D | simple_propagator_state.h | 121 Entry* GetInputTensors(const TaggedNode& tagged_node) { 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::__anone4efb82c0111::SamplingDatasetParams
|
| D | random_dataset_op_test.cc | 56 virtual std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon1373acb20111::RandomDatasetParams
|
| D | assert_next_dataset_op_test.cc | 42 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon2a326c810111::AssertNextDatasetParams
|
| D | unique_dataset_op_test.cc | 38 std::vector<Tensor> GetInputTensors() const override { return {}; } in GetInputTensors() function in tensorflow::data::experimental::__anon085fcb880111::UniqueDatasetParams
|
| D | lmdb_dataset_op_test.cc | 44 std::vector<Tensor> GetInputTensors() const override { return {filenames_}; } in GetInputTensors() function in tensorflow::data::experimental::__anon5019c0b00111::LMDBDatasetParams
|
| D | assert_prev_dataset_op_test.cc | 64 std::vector<Tensor> GetInputTensors() const override { in GetInputTensors() function in tensorflow::data::experimental::__anon21b5395f0111::AssertPrevDatasetParams
|