| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unpack_test.cc | 58 std::vector<std::vector<int>> output_shapes; in GetOutputShapes() local
77 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
97 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
115 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
133 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
153 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
169 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
188 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
209 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
228 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local
[all …]
|
| /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 | 116 std::vector<PartialTensorShape> output_shapes; in __anon22ceab0f0202() local
145 std::vector<PartialTensorShape> output_shapes; in __anon22ceab0f0302() local
205 std::vector<PartialTensorShape> output_shapes; in __anon22ceab0f0502() local
|
| D | dataset_ops.cc | 525 std::vector<PartialTensorShape> output_shapes; in IteratorGetNextShapeFn() local
679 std::vector<PartialTensorShape> output_shapes; in __anondee5e0a21402() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.h | 38 const std::vector<PartialTensorShape>& output_shapes) in GraphRewriteDataset()
58 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function
|
| D | window_dataset.cc | 26 std::vector<PartialTensorShape> output_shapes) in WindowDataset()
40 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon8ba9a6530111::WindowDataset
112 std::vector<PartialTensorShape> output_shapes, in NewWindowDataset()
|
| D | generator_dataset_op.cc | 37 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
53 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::GeneratorDatasetOp::Dataset
|
| D | filter_by_component_dataset_op.cc | 55 std::vector<PartialTensorShape> output_shapes) in Dataset()
74 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anond0e1a64b0111::FilterByLastComponentDatasetOp::Dataset
|
| D | take_dataset_op.h | 49 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function
|
| D | window_dataset_op.cc | 90 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon256b48110111::WindowDatasetOp::Dataset
91 static std::vector<PartialTensorShape>* output_shapes = in output_shapes() local
238 std::vector<PartialTensorShape> output_shapes( in GetNextInternal() local
|
| D | flat_map_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::__anond045031d0111::FlatMapDatasetOp::Dataset
|
| D | optimize_dataset_op.cc | 64 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | iterator_ops.py | 75 output_shapes, output_classes): argument
112 output_shapes=None, argument
222 output_shapes=None, argument
456 def output_shapes(self): member in Iterator
602 def output_shapes(self): member in EagerIterator
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_scan_dataset_op.cc | 73 std::vector<PartialTensorShape> output_shapes; in MakeDataset() local
96 std::vector<PartialTensorShape> output_shapes) in Dataset()
124 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon558a87df0111::BigtableScanDatasetOp::Dataset
|
| D | bigtable_lookup_dataset_op.cc | 43 std::vector<PartialTensorShape> output_shapes; in MakeDataset() local
65 std::vector<PartialTensorShape> output_shapes) in Dataset()
93 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon13b314540111::BigtableLookupDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | map_defun.py | 29 output_shapes, argument
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | script_ops.py | 37 output_shapes=None, argument
|
| /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
| D | trt_shfn.cc | 58 std::vector<tensorflow::TensorShape> output_shapes; in TRTEngineOpShapeInference() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | non_serializable_dataset_op.cc | 44 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
63 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon4d42c3f70111::NonSerializableDatasetOp::Dataset
|
| D | assert_next_dataset_op.cc | 51 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
71 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon9dfb5fa60111::AssertNextDatasetOp::Dataset
|
| D | indexed_dataset_op.cc | 105 const std::vector<PartialTensorShape>& output_shapes) in MaterializedDatasetResource()
128 const std::vector<PartialTensorShape>& output_shapes() const { in output_shapes() function in tensorflow::data::__anon0a1add170111::MaterializedDatasetResource
435 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon0a1add170111::IdentityIndexedDatasetOp::Dataset
505 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon0a1add170111::IdentityIndexedDatasetOp::Dataset::Materialized
|
| D | sql_dataset_op.cc | 84 const std::vector<PartialTensorShape>& output_shapes) in Dataset()
102 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon6e0d6b260111::SqlDatasetOp::Dataset
|
| D | choose_fastest_branch_dataset_op.cc | 39 const std::vector<PartialTensorShape>* output_shapes, in WrapperDataset()
50 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anonf4b3f19d0111::WrapperDataset
193 const std::vector<PartialTensorShape>& output_shapes, in Dataset()
221 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anonf4b3f19d0111::ChooseFastestBranchDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util_test.cc | 45 std::vector<PartialTensorShape> output_shapes; in TEST() local
|
| /external/tensorflow/tensorflow/contrib/ignite/kernels/dataset/ |
| D | ignite_dataset.cc | 68 const std::vector<PartialTensorShape>& IgniteDataset::output_shapes() const { in output_shapes() function in tensorflow::IgniteDataset
|