| /external/tensorflow/tensorflow/lite/kernels/ctc/ |
| 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
139 std::vector<PartialTensorShape> output_shapes; in CaseShapeInferenceFn() local
211 std::vector<PartialTensorShape> output_shapes; in WhileShapeInferenceFn() local
|
| D | tpu_embedding_ops.cc | 75 std::vector<TensorShapeProto> output_shapes; in __anonc21aa7d60102() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 101 output_shapes, output_classes): argument
142 output_shapes=None, argument
237 output_shapes=None, argument
485 def output_shapes(self): member in Iterator
810 def output_shapes(self): member in OwnedIterator
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| 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::__anon1e3d86f60111::Window
185 std::vector<PartialTensorShape> output_shapes, in NewWindow()
|
| 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 | 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 | 59 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
78 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 | get_options_op_test.cc | 36 std::vector<PartialTensorShape> output_shapes, in GetOptionsParams()
|
| /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::__anon1408725a0111::NonSerializableDatasetOp::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::__anon872515e00111::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::__anon872515e00111::ChooseFastestBranchDatasetOp::Dataset
|
| 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
|
| 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
|
| 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::__anon265533720111::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::__anon265533720111::RebatchDatasetV2Op::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::__anon43d1fca90111::SqlDatasetOp::Dataset
|
| 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::__anon3435af1e0111::ScanDatasetOp::Dataset
|
| 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()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.h | 145 std::vector<PartialTensorShape> output_shapes() const { in output_shapes() function
220 std::vector<PartialTensorShape> output_shapes, in BatchDatasetParams()
257 std::vector<PartialTensorShape> output_shapes, in MapDatasetParams()
328 std::vector<PartialTensorShape> output_shapes, in TakeDatasetParams()
358 std::vector<PartialTensorShape> output_shapes, in ConcatenateDatasetParams()
387 std::vector<PartialTensorShape> output_shapes, in OptionsDatasetParams()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 45 std::vector<PartialTensorShape> output_shapes; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reduce_scatter_combiner.cc | 68 std::vector<Shape> output_shapes; in CombineReduceScatters() local
|
| D | all_gather_combiner.cc | 62 std::vector<Shape> output_shapes; in CombineAllGathers() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | node_shader.h | 108 std::vector<std::array<int64_t, 4>> output_shapes; member
|