| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/ |
| H A D | ctc_beam_search_decoder_test.cc | 84 std::vector<std::vector<int>> output_shapes; in GetOutputShapes() local
105 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
137 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
169 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
210 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | iterator_ops.py | 102 output_shapes, output_classes): argument
150 output_shapes=None, argument
245 output_shapes=None, argument
497 def output_shapes(self): member in Iterator
810 def output_shapes(self): member in OwnedIterator
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | functional_ops.cc | 94 std::vector<PartialTensorShape> output_shapes; in IfShapeInferenceFn() local
139 std::vector<PartialTensorShape> output_shapes; in CaseShapeInferenceFn() local
211 std::vector<PartialTensorShape> output_shapes; in WhileShapeInferenceFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_test_base.h | 153 std::vector<PartialTensorShape> output_shapes() const { in output_shapes() function
228 std::vector<PartialTensorShape> output_shapes, in BatchDatasetParams()
265 std::vector<PartialTensorShape> output_shapes, in MapDatasetParams()
338 std::vector<PartialTensorShape> output_shapes, in TakeDatasetParams()
368 std::vector<PartialTensorShape> output_shapes, in ConcatenateDatasetParams()
397 std::vector<PartialTensorShape> output_shapes, in OptionsDatasetParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | choose_fastest_branch_dataset_op.cc | 41 const std::vector<PartialTensorShape>* output_shapes, in WrapperDataset()
52 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon50b30f2a0111::WrapperDataset
207 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
234 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon50b30f2a0111::ChooseFastestBranchDatasetOp::Dataset
|
| H A D | rebatch_dataset_op.cc | 59 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
84 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetOp::Dataset
322 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
347 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetV2Op::Dataset
|
| H A D | non_serializable_dataset_op.cc | 45 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
64 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon6c78f5640111::NonSerializableDatasetOp::Dataset
|
| H A D | assert_next_dataset_op.cc | 39 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
57 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::AssertNextDatasetOp::Dataset
|
| H A D | assert_cardinality_dataset_op.cc | 43 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
61 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::AssertCardinalityDatasetOp::Dataset
|
| H A D | sql_dataset_op.cc | 82 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
100 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anonaddcc7730111::SqlDatasetOp::Dataset
|
| H A D | list_dataset_op.cc | 47 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
72 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::ListDatasetOp::Dataset
|
| H A D | threadpool_dataset_op.cc | 191 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::ThreadPoolDatasetOp::Dataset
313 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::MaxIntraOpParallelismDatasetOp::Dataset
454 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::PrivateThreadPoolDatasetOp::Dataset
|
| H A D | assert_prev_dataset_op.cc | 138 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
156 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::AssertPrevDatasetOp::Dataset
|
| H A D | scan_dataset_op.cc | 78 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
103 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anondf99d3280111::ScanDatasetOp::Dataset
|
| H A D | parallel_interleave_dataset_op_test.cc | 35 const std::vector<PartialTensorShape>& output_shapes, string node_name) in ParallelInterleaveDatasetParams()
116 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
|
| H A D | io_ops.cc | 240 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::SaveDatasetV2Op::Dataset
560 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
577 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::LoadDatasetOp::Dataset
|
| H A D | group_by_reducer_dataset_op.cc | 85 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
108 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anona8bc84f30111::GroupByReducerDatasetOp::Dataset
|
| H A D | stats_dataset_ops.cc | 71 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon296061c50111::LatencyStatsDatasetOp::Dataset
196 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon296061c50111::BytesProducedStatsDatasetOp::Dataset
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | window_dataset.cc | 39 std::vector<PartialTensorShape> output_shapes) in Window()
53 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon2a1301800111::Window
185 std::vector<PartialTensorShape> output_shapes, in NewWindow()
|
| H A D | parallel_map_dataset_op.cc | 79 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
89 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
117 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::ParallelMapDatasetOp::Dataset
787 const std::vector<PartialTensorShape>& output_shapes) { in MakeDataServiceUncompressDataset()
|
| H A D | generator_dataset_op.cc | 54 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
70 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::GeneratorDatasetOp::Dataset
|
| H A D | map_dataset_op.cc | 47 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
68 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::MapDatasetOp::Dataset
|
| H A D | flat_map_dataset_op.cc | 61 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
80 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::FlatMapDatasetOp::Dataset
|
| H A D | interleave_dataset_op_test.cc | 32 std::vector<PartialTensorShape> output_shapes, in InterleaveDatasetParams()
99 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | input_lib.py | 206 def output_shapes(self): member in DistributedIteratorV1
364 def output_shapes(self): member in _SingleWorkerDatasetIterator
|