/external/tensorflow/tensorflow/lite/kernels/ |
D | unpack_test.cc | 58 std::vector<std::vector<int>> output_shapes; in GetOutputShapes() local 60 output_shapes.push_back(GetTensorShape(output)); in GetOutputShapes() 62 return output_shapes; in GetOutputShapes() 77 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local 78 EXPECT_EQ(output_shapes.size(), 3); in TEST() 79 EXPECT_THAT(output_shapes[0], ElementsAre(2)); in TEST() 80 EXPECT_THAT(output_shapes[1], ElementsAre(2)); in TEST() 81 EXPECT_THAT(output_shapes[2], ElementsAre(2)); in TEST() 97 const std::vector<std::vector<int>>& output_shapes = model.GetOutputShapes(); in TEST() local 98 EXPECT_EQ(output_shapes.size(), 2); in TEST() [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 86 output_shapes.push_back(GetTensorShape(output)); in GetOutputShapes() 88 return output_shapes; in GetOutputShapes() 105 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local 106 EXPECT_EQ(output_shapes.size(), 4); in TEST() 107 EXPECT_THAT(output_shapes[0], ElementsAre(1, 2)); in TEST() 108 EXPECT_THAT(output_shapes[1], ElementsAre(1)); in TEST() 109 EXPECT_THAT(output_shapes[2], ElementsAre(2)); in TEST() 110 EXPECT_THAT(output_shapes[3], ElementsAre(1, 1)); in TEST() 137 const std::vector<std::vector<int>>& output_shapes = m.GetOutputShapes(); in TEST() local [all …]
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | iterator_ops.py | 75 output_shapes, output_classes): argument 97 if (output_types is None or output_shapes is None 103 output_types, output_shapes, output_classes) 112 output_shapes=None, argument 181 if output_shapes is None: 182 output_shapes = nest.map_structure( 185 output_shapes = nest.map_structure_up_to( 186 output_types, tensor_shape.as_shape, output_shapes) 189 nest.assert_same_structure(output_types, output_shapes) 191 output_types, output_shapes, output_classes) [all …]
|
D | dataset_ops.py | 243 output_shapes = nest.map_structure(str, get_legacy_output_shapes(self)) 244 output_shapes = str(output_shapes).replace("'", "") 247 return ("<%s shapes: %s, types: %s>" % (type(self).__name__, output_shapes, 328 def from_generator(generator, output_types, output_shapes=None, args=None): argument 390 if output_shapes is None: 391 output_shapes = nest.map_structure( 394 output_shapes = nest.map_structure_up_to( 395 output_types, tensor_shape.as_shape, output_shapes) 402 flattened_shapes = nest.flatten(output_shapes) 485 if output_shapes is not None: [all …]
|
/external/tensorflow/tensorflow/core/ops/ |
D | functional_ops.cc | 116 std::vector<PartialTensorShape> output_shapes; in __anon22ceab0f0202() local 117 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in __anon22ceab0f0202() 120 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in __anon22ceab0f0202() 121 if (output_shapes.size() != c->num_outputs()) { in __anon22ceab0f0202() 124 output_shapes.size(), " vs. ", c->num_outputs()); in __anon22ceab0f0202() 126 for (size_t i = 0; i < output_shapes.size(); ++i) { in __anon22ceab0f0202() 129 output_shapes[i], &output_shape_handle)); in __anon22ceab0f0202() 145 std::vector<PartialTensorShape> output_shapes; in __anon22ceab0f0302() local 146 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in __anon22ceab0f0302() 149 if (output_shapes.empty()) return shape_inference::UnknownShape(c); in __anon22ceab0f0302() [all …]
|
D | dataset_ops.cc | 525 std::vector<PartialTensorShape> output_shapes; in IteratorGetNextShapeFn() local 526 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in IteratorGetNextShapeFn() 527 if (output_shapes.size() != c->num_outputs()) { in IteratorGetNextShapeFn() 530 output_shapes.size(), " vs. ", c->num_outputs()); in IteratorGetNextShapeFn() 532 for (size_t i = 0; i < output_shapes.size(); ++i) { in IteratorGetNextShapeFn() 535 output_shapes[i], &output_shape_handle)); in IteratorGetNextShapeFn() 679 std::vector<PartialTensorShape> output_shapes; in __anondee5e0a21402() local 680 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in __anondee5e0a21402() 683 if (output_shapes.size() != c->num_outputs()) { in __anondee5e0a21402() 686 output_shapes.size(), " vs. ", c->num_outputs(), ")"); in __anondee5e0a21402() [all …]
|
/external/tensorflow/tensorflow/contrib/framework/python/ops/ |
D | script_ops.py | 37 output_shapes=None, argument 109 if callable(output_shapes): 111 output_shapes = output_shapes(*args, **kwargs) 134 if output_shapes is not None: 136 output_shapes = nest.map_structure_up_to( 137 output_types, tensor_shape.as_shape, output_shapes) 139 flattened_shapes = nest.flatten(output_shapes)
|
/external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
D | trt_shfn.cc | 58 std::vector<tensorflow::TensorShape> output_shapes; in TRTEngineOpShapeInference() local 59 TF_RETURN_IF_ERROR(c->GetAttr("output_shapes", &output_shapes)); in TRTEngineOpShapeInference() 60 if (output_shapes.size() != c->num_outputs()) { in TRTEngineOpShapeInference() 64 c->num_outputs(), " vs ", output_shapes.size()); in TRTEngineOpShapeInference() 66 for (size_t i = 0; i < output_shapes.size(); ++i) { in TRTEngineOpShapeInference() 69 c->MakeShapeFromTensorShape(output_shapes.at(i), &handle)); in TRTEngineOpShapeInference()
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | map_defun.py | 29 output_shapes, argument 57 if not isinstance(output_shapes, list): 65 output_shapes = [tensor_shape.TensorShape(s) for s in output_shapes] 67 output_dtypes, output_shapes, concrete_fn,
|
D | batching.py | 484 output_shapes=None, argument 528 if output_shapes is None: 530 output_shapes = nest.pack_sequence_as( 535 nest.assert_same_structure(output_types, output_shapes) 537 flat_new_shapes = nest.flatten_up_to(output_types, output_shapes) 545 output_shapes)) 546 output_shapes = nest.map_structure_up_to( 547 output_types, tensor_shape.as_shape, output_shapes) 556 output_types, output_shapes, output_classes) 737 def recalculate_output_shapes(output_shapes): argument [all …]
|
D | indexed_dataset_ops.py | 36 output_types, output_shapes): argument 41 self._output_shapes = output_shapes 64 output_shapes=nest.flatten( 104 self.output_shapes)
|
D | interleave_ops.py | 114 output_shapes = dataset_ops.get_legacy_output_shapes(self._data_inputs[0]) 116 output_shapes = nest.pack_sequence_as(output_shapes, [ 118 nest.flatten(output_shapes), 123 first_output_types, output_shapes, first_output_classes)
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | encapsulate_util_test.cc | 45 std::vector<PartialTensorShape> output_shapes; in TEST() local 47 &output_shapes)); in TEST() 48 EXPECT_EQ(output_shapes.size(), 1); in TEST() 50 output_shapes[0].AsProto(&shape_proto); in TEST()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | window_dataset.cc | 26 std::vector<PartialTensorShape> output_shapes) in WindowDataset() argument 30 output_shapes_(std::move(output_shapes)) {} 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() argument 117 std::move(output_shapes)); in NewWindowDataset()
|
D | graph_rewrite_dataset.h | 38 const std::vector<PartialTensorShape>& output_shapes) in GraphRewriteDataset() argument 43 output_shapes_(output_shapes) { in GraphRewriteDataset() 58 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function
|
D | map_defun_op.cc | 144 std::vector<PartialTensorShape> output_shapes GUARDED_BY(mu); 159 output_shapes(output_shapes_attr) { in ComputeOptions() 221 if (!compute_opts_->output_shapes.at(index).IsCompatibleWith( in SetRetval() 226 compute_opts_->output_shapes.at(index).DebugString(), "."); in SetRetval() 228 if (!compute_opts_->output_shapes.at(index).IsFullyDefined()) { in SetRetval() 231 compute_opts_->output_shapes.at(index) = val.shape(); in SetRetval()
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.data.-iterator.pbtxt | 15 name: "output_shapes" 24 …\'self\', \'iterator_resource\', \'initializer\', \'output_types\', \'output_shapes\', \'output_cl… 28 …argspec: "args=[\'string_handle\', \'output_types\', \'output_shapes\', \'output_classes\'], varar… 32 …argspec: "args=[\'output_types\', \'output_shapes\', \'shared_name\', \'output_classes\'], varargs…
|
/external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
D | bigtable_lookup_dataset_op.cc | 43 std::vector<PartialTensorShape> output_shapes; in MakeDataset() local 44 output_shapes.reserve(num_outputs); in MakeDataset() 48 output_shapes.push_back({}); in MakeDataset() 54 std::move(columns), output_types, std::move(output_shapes)); in MakeDataset() 65 std::vector<PartialTensorShape> output_shapes) in Dataset() argument 72 output_shapes_(std::move(output_shapes)), in Dataset() 93 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon13b314540111::BigtableLookupDatasetOp::Dataset
|
D | bigtable_scan_dataset_op.cc | 73 std::vector<PartialTensorShape> output_shapes; in MakeDataset() local 74 output_shapes.reserve(num_outputs); in MakeDataset() 78 output_shapes.push_back({}); in MakeDataset() 85 probability, output_types, std::move(output_shapes)); in MakeDataset() 96 std::vector<PartialTensorShape> output_shapes) in Dataset() argument 108 output_shapes_(std::move(output_shapes)) { in Dataset() 124 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon558a87df0111::BigtableScanDatasetOp::Dataset
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | indexed_dataset_op.cc | 54 virtual const std::vector<PartialTensorShape>& output_shapes() const = 0; 105 const std::vector<PartialTensorShape>& output_shapes) in MaterializedDatasetResource() argument 106 : output_dtypes_(output_dtypes), output_shapes_(output_shapes) {} in MaterializedDatasetResource() 128 const std::vector<PartialTensorShape>& output_shapes() const { in output_shapes() function in tensorflow::data::__anon0a1add170111::MaterializedDatasetResource 138 VerifyShapesCompatible(output_shapes_, dataset->output_shapes())); in set_materialized_dataset() 314 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource() 377 auto expected_shapes = materialized_resource->output_shapes(); in Compute() 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 506 return dataset_->output_shapes(); in output_shapes()
|
D | choose_fastest_dataset_op.cc | 69 ctx, inputs[i]->output_shapes().size() == output_shapes_.size(), in MakeDataset() 73 i, " has ", inputs[i]->output_shapes().size(), in MakeDataset() 81 .MergeWith(inputs[i]->output_shapes().at(j), &result) in MakeDataset() 87 " has shape: ", inputs[i]->output_shapes().at(j), in MakeDataset() 120 const std::vector<PartialTensorShape>& output_shapes, in Dataset() argument 125 output_shapes_(output_shapes), in Dataset() 150 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon579bbc700111::ChooseFastestDatasetOp::Dataset
|
D | directed_interleave_dataset_op.cc | 41 selector_input->output_shapes().size() == 1 && in MakeDataset() 42 selector_input->output_shapes()[0].IsCompatibleWith( in MakeDataset() 75 output_shapes_ = data_inputs_[0]->output_shapes(); in Dataset() 82 output_shapes_[j], data_input->output_shapes()[j]); in Dataset() 104 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon9a3736c80111::DirectedInterleaveDatasetOp::Dataset
|
D | non_serializable_dataset_op.cc | 44 const std::vector<PartialTensorShape>& output_shapes) in Dataset() argument 48 output_shapes_(output_shapes) { in Dataset() 63 const std::vector<PartialTensorShape>& output_shapes() const override { in output_shapes() function in tensorflow::data::__anon4d42c3f70111::NonSerializableDatasetOp::Dataset
|
/external/tensorflow/tensorflow/python/data/kernel_tests/ |
D | from_generator_test.py | 107 output_shapes=([None], [3])) 160 generator, output_types=dtypes.int64, output_shapes=[]).prefetch(2) 180 generator, output_types=dtype, output_shapes=[1]) 197 generator, output_types=dtypes.string, output_shapes=[]) 209 generator, output_types=dtypes.int64, output_shapes=[3]) 229 generator, output_types=dtypes.int64, output_shapes=[3]) 325 generator_with_arg, output_types=dtypes.int64, output_shapes=(), 342 output_shapes=((), ()), args=(elem, message))
|
/external/tensorflow/tensorflow/contrib/data/python/ops/ |
D | batching.py | 219 output_shapes = _merge_output_shapes( 225 output_shapes=output_shapes,
|