| /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
139 std::vector<PartialTensorShape> output_shapes; in CaseShapeInferenceFn() local
211 std::vector<PartialTensorShape> output_shapes; in WhileShapeInferenceFn() local
|
| D | tpu_embedding_ops.cc | 73 std::vector<TensorShapeProto> output_shapes; in __anon355f22aa0102() 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::__anonf4afd0ca0111::WindowDataset
133 std::vector<PartialTensorShape> output_shapes, in NewWindowDataset()
|
| D | dataset_test_base.h | 145 std::vector<PartialTensorShape> output_shapes() const { in output_shapes() function
211 std::vector<PartialTensorShape> output_shapes, in BatchDatasetParams()
248 std::vector<PartialTensorShape> output_shapes, in MapDatasetParams()
319 std::vector<PartialTensorShape> output_shapes, in TakeDatasetParams()
349 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 | 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 | 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 | interleave_dataset_op_test.cc | 32 std::vector<PartialTensorShape> output_shapes, in InterleaveDatasetParams()
98 const std::vector<PartialTensorShape>& output_shapes) { in MakeTensorSliceDatasetFunc()
|
| D | optimize_dataset_op_test.cc | 30 std::vector<PartialTensorShape> output_shapes, in OptimizeDatasetParams()
|
| D | interleave_dataset_op.cc | 56 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
82 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 | 84 output_shapes, output_classes): argument
125 output_shapes=None, argument
220 output_shapes=None, argument
467 def output_shapes(self): member in Iterator
770 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::__anon6ce33d2e0111::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::__anon98d3c5340111::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::__anon98d3c5340111::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::__anon6916dfc60111::RebatchDatasetOp::Dataset
316 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
341 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon6916dfc60111::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::__anone5229efd0111::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::__anonff9a9bf20111::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()
|
| D | threadpool_dataset_op.cc | 166 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon16b1edff0111::ThreadPoolDatasetOp::Dataset
296 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon16b1edff0111::MaxIntraOpParallelismDatasetOp::Dataset
421 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::experimental::__anon16b1edff0111::PrivateThreadPoolDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 45 std::vector<PartialTensorShape> output_shapes; in TEST() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | node_shader.h | 108 std::vector<std::array<int64_t, 4>> output_shapes; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 117 std::vector<Shape> output_shapes; in TEST_F() local
|