| /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 | 72 std::vector<TensorShapeProto> output_shapes; in __anon3e675d600102() local
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| 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
|
| /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::__anon0f62bc000111::Window
185 std::vector<PartialTensorShape> output_shapes, in NewWindow()
|
| 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
|
| 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
|
| D | rewrite_dataset_op_test.cc | 30 std::vector<PartialTensorShape> output_shapes, in RewriteDatasetParams()
|
| 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
|
| D | optimize_dataset_op_test.cc | 30 std::vector<PartialTensorShape> output_shapes, in OptimizeDatasetParams()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 25 output_shapes, argument
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ops/ |
| D | trt_engine_op.cc | 49 std::vector<tensorflow::PartialTensorShape> output_shapes; in __anon514a20330102() local
|
| /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::__anon75f8cfe40111::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::__anond0bf79aa0111::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::__anond0bf79aa0111::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::__anon08085dbc0111::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::__anon08085dbc0111::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::__anonc58791f30111::SqlDatasetOp::Dataset
|
| 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
|
| 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
|
| /external/executorch/devtools/debug_format/ |
| D | base_schema.py | 27 output_shapes: Optional[List[List[int]]] = None variable in Node
|
| /external/tensorflow/tensorflow/core/data/ |
| 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()
|
| /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 | 67 std::vector<const Shape*> output_shapes; in CombineReduceScatters() local
|
| D | all_gather_combiner.cc | 61 std::vector<const Shape*> output_shapes; in CombineAllGathers() local
|