| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| 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
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | functional_ops.cc | 94 std::vector<PartialTensorShape> output_shapes; in IfShapeInferenceFn() local
148 std::vector<PartialTensorShape> output_shapes; in __anon4a8e12910202() local
199 std::vector<PartialTensorShape> output_shapes; in WhileShapeInferenceFn() local
|
| D | dataset_ops.cc | 630 std::vector<PartialTensorShape> output_shapes; in IteratorGetNextShapeFn() local
800 std::vector<PartialTensorShape> output_shapes; in __anonc0e8a7641a02() local
|
| D | tpu_embedding_ops.cc | 137 std::vector<TensorShapeProto> output_shapes; in __anon4b0f15b50602() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset.cc | 32 std::vector<PartialTensorShape> output_shapes) in WindowDataset()
46 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon280aa0150111::WindowDataset
128 std::vector<PartialTensorShape> output_shapes, in NewWindowDataset()
|
| D | iterator_ops.h | 35 const std::vector<PartialTensorShape>& output_shapes, in IteratorResource()
66 const std::vector<PartialTensorShape>& output_shapes() const { in output_shapes() function
|
| D | generator_dataset_op.cc | 53 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
69 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::GeneratorDatasetOp::Dataset
|
| D | dataset_test_base.h | 148 std::vector<PartialTensorShape> output_shapes() const { in output_shapes() function
214 std::vector<PartialTensorShape> output_shapes, in BatchDatasetParams()
251 std::vector<PartialTensorShape> output_shapes, in MapDatasetParams()
322 std::vector<PartialTensorShape> output_shapes, in TakeDatasetParams()
352 std::vector<PartialTensorShape> output_shapes, in ConcatenateDatasetParams()
|
| D | map_dataset_op.cc | 46 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
67 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::MapDatasetOp::Dataset
|
| D | flat_map_dataset_op.cc | 51 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
70 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::FlatMapDatasetOp::Dataset
|
| D | optimize_dataset_op_test.cc | 30 std::vector<PartialTensorShape> output_shapes, in OptimizeDatasetParams()
|
| D | interleave_dataset_op_test.cc | 32 std::vector<PartialTensorShape> output_shapes, in InterleaveDatasetParams()
98 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
|
| D | interleave_dataset_op.cc | 55 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
76 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::InterleaveDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 80 output_shapes, output_classes): argument
121 output_shapes=None, argument
226 output_shapes=None, argument
464 def output_shapes(self): member in Iterator
693 def output_shapes(self): member in OwnedIterator
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 29 output_shapes, argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| 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::__anon2b049c390111::NonSerializableDatasetOp::Dataset
|
| D | assert_next_dataset_op.cc | 38 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
56 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::AssertNextDatasetOp::Dataset
|
| 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::__anon0751431f0111::WrapperDataset
200 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
227 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon0751431f0111::ChooseFastestBranchDatasetOp::Dataset
|
| 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::__anon8526b0280111::SqlDatasetOp::Dataset
|
| D | rebatch_dataset_op.cc | 56 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
78 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anona1c58e710111::RebatchDatasetOp::Dataset
|
| D | scan_dataset_op.cc | 79 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
104 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anona020d27d0111::ScanDatasetOp::Dataset
|
| D | parallel_interleave_dataset_op_test.cc | 34 std::vector<PartialTensorShape> output_shapes, string node_name) in ParallelInterleaveDatasetParams()
113 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
|
| D | auto_shard_dataset_op_test.cc | 29 std::vector<PartialTensorShape> output_shapes, in AutoShardDatasetParams()
|
| D | threadpool_dataset_op.cc | 165 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon1867a0ca0111::ThreadPoolDatasetOp::Dataset
276 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon1867a0ca0111::MaxIntraOpParallelismDatasetOp::Dataset
380 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon1867a0ca0111::PrivateThreadPoolDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 45 std::vector<PartialTensorShape> output_shapes; in TEST() local
|