Home
last modified time | relevance | path

Searched refs:output_shapes (Results 1 – 25 of 191) sorted by relevance

12345678

/external/tensorflow/tensorflow/lite/kernels/
Dunpack_test.cc58 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/
Dctc_beam_search_decoder_test.cc84 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/
Diterator_ops.py75 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 …]
Ddataset_ops.py243 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/
Dfunctional_ops.cc116 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 …]
Ddataset_ops.cc525 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/
Dscript_ops.py37 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/
Dtrt_shfn.cc58 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/
Dmap_defun.py29 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,
Dbatching.py484 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 …]
Dindexed_dataset_ops.py36 output_types, output_shapes): argument
41 self._output_shapes = output_shapes
64 output_shapes=nest.flatten(
104 self.output_shapes)
Dinterleave_ops.py114 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/
Dencapsulate_util_test.cc45 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/
Dwindow_dataset.cc26 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()
Dgraph_rewrite_dataset.h38 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
Dmap_defun_op.cc144 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/
Dtensorflow.data.-iterator.pbtxt15 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/
Dbigtable_lookup_dataset_op.cc43 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
Dbigtable_scan_dataset_op.cc73 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/
Dindexed_dataset_op.cc54 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()
Dchoose_fastest_dataset_op.cc69 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
Ddirected_interleave_dataset_op.cc41 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
Dnon_serializable_dataset_op.cc44 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/
Dfrom_generator_test.py107 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/
Dbatching.py219 output_shapes = _merge_output_shapes(
225 output_shapes=output_shapes,

12345678