Searched defs:output_shapes (Results 1 - 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/
H A Ddataset_ops.cc39 .Attr("output_shapes: list(shape) >= 1")
45 // `output_shapes`.
51 .Attr("output_shapes: list(shape) >= 1")
57 // `output_shapes`.
73 .Attr("output_shapes: list(shape) >= 1")
82 .Attr("output_shapes: list(shape) >= 1")
90 .Attr("output_shapes: list(shape) >= 1")
99 .Attr("output_shapes: list(shape) >= 1")
107 .Attr("output_shapes: list(shape) >= 1")
115 .Attr("output_shapes
410 std::vector<PartialTensorShape> output_shapes; local
[all...]
/external/tensorflow/tensorflow/contrib/data/python/ops/
H A Dunique.py67 output_shapes=nest.flatten(
68 sparse.as_dense_shapes(self.output_shapes, self.output_classes)),
77 def output_shapes(self): member in class:UniqueDataset
78 return self._input_dataset.output_shapes
H A Derror_ops.py67 output_shapes=nest.flatten(
68 sparse.as_dense_shapes(self.output_shapes, self.output_classes)),
77 def output_shapes(self): member in class:IgnoreErrorsDataset
78 return self._input_dataset.output_shapes
H A Drandom_ops.py52 output_shapes=nest.flatten(
53 sparse.as_dense_shapes(self.output_shapes, self.output_classes)),
62 def output_shapes(self): member in class:RandomDataset
H A Dreaders.py194 nest.flatten(self.output_shapes))
201 def output_shapes(self): member in class:SqlDataset
H A Dscan_ops.py78 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
157 output_shapes=nest.flatten(
158 sparse.as_dense_shapes(self.output_shapes, self.output_classes)))
165 def output_shapes(self): member in class:_ScanDataset
H A Dshuffle_ops.py71 output_shapes=nest.flatten(
72 sparse.as_dense_shapes(self.output_shapes, self.output_classes)))
80 def output_shapes(self): member in class:_ShuffleAndRepeatDataset
81 return self._input_dataset.output_shapes
H A Dgrouping.py46 (having shapes and types defined by `self.output_shapes` and
91 def __init__(self, dataset_variant, output_types, output_shapes,
96 self._output_shapes = output_shapes
107 def output_shapes(self): member in class:_VariantDataset
153 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
160 nested_args, input_dataset.output_types, input_dataset.output_shapes,
185 input_dataset.output_shapes, input_dataset.output_classes)
193 self._output_shapes = output_dataset.output_shapes
204 def output_shapes(self): member in class:GroupByWindowDataset
222 output_shapes
[all...]
H A Dinterleave_ops.py46 dense_shapes = sparse.as_dense_shapes(input_dataset.output_shapes,
53 nested_args, input_dataset.output_types, input_dataset.output_shapes,
65 self._output_shapes = dataset.output_shapes
99 output_shapes=nest.flatten(
100 sparse.as_dense_shapes(self.output_shapes, self.output_classes)))
107 def output_shapes(self): member in class:ParallelInterleaveDataset
211 and types defined by `self.output_shapes` and `self.output_types`) to a
H A Dstats_ops.py167 output_shapes=nest.flatten(
168 sparse.as_dense_shapes(self.output_shapes, self.output_classes)))
171 def output_shapes(self): member in class:_StatsDataset
172 return self._input_dataset.output_shapes
H A Dbatching.py140 dataset.output_shapes)
164 print(batched.output_shapes) # ==> "(128,)" (the batch dimension is known)
242 output_shapes=nest.flatten(
243 sparse.as_dense_shapes(self.output_shapes, self.output_classes)),
252 def output_shapes(self): member in class:DenseToSparseBatchDataset
266 output_shapes=None,
272 * Each shape in `dataset.output_shapes` must be compatible with each shape
273 in `output_shapes` (if given).
281 output_shapes: (Optional.) A nested structure of `tf.TensorShape` objects.
287 ValueError: If either `output_types` or `output_shapes` i
344 def output_shapes(self): member in class:_RestructuredDataset
375 def output_shapes(self): member in class:_MapAndBatchDataset
[all...]
H A Ddataset_ops.py55 def output_shapes(self): member in class:Dataset
56 return self._dataset.output_shapes
105 def from_generator(generator, output_types, output_shapes=None):
111 `output_types` and (optional) `output_shapes` arguments.
135 output_shapes: (Optional.) A nested structure of `tf.TensorShape`
143 generator, output_types, output_shapes))
489 shapes and types defined by `self.output_shapes` and
522 and types defined by `self.output_shapes` and `self.output_types`) to a
587 and types defined by `self.output_shapes` and `self.output_types`) to a
612 and types defined by `self.output_shapes` an
[all...]
/external/tensorflow/tensorflow/contrib/kafka/python/ops/
H A Dkafka_dataset_ops.py69 def output_shapes(self): member in class:KafkaDataset
/external/tensorflow/tensorflow/core/kernels/data/
H A Dwindow_dataset.cc24 std::vector<PartialTensorShape> output_shapes)
27 output_shapes_(std::move(output_shapes)) {}
37 const std::vector<PartialTensorShape>& output_shapes() const override {
90 std::vector<PartialTensorShape> output_shapes,
93 // the elements match the output_types and output_shapes.
95 std::move(output_shapes));
22 WindowDataset(std::vector<std::vector<Tensor>> elements, DataTypeVector output_types, std::vector<PartialTensorShape> output_shapes) argument
88 NewWindowDataset(std::vector<std::vector<Tensor>> elements, DataTypeVector output_types, std::vector<PartialTensorShape> output_shapes, DatasetBase** out_dataset) argument
H A Dsql_dataset_ops.cc36 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
82 const std::vector<PartialTensorShape>& output_shapes)
87 output_shapes_(output_shapes) {}
99 const std::vector<PartialTensorShape>& output_shapes() const override {
80 Dataset(const string& driver_name, const string& data_source_name, const string& query, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dmap_dataset_op.cc36 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
64 const std::vector<PartialTensorShape>& output_shapes)
70 output_shapes_(output_shapes) {
85 const std::vector<PartialTensorShape>& output_shapes() const override {
60 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& func, std::unique_ptr<CapturedFunction> captured_func, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dflat_map_dataset_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
65 const std::vector<PartialTensorShape>& output_shapes)
71 output_shapes_(output_shapes) {
87 const std::vector<PartialTensorShape>& output_shapes() const override {
61 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& func, std::unique_ptr<CapturedFunction> captured_func, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dinterleave_dataset_op.cc38 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
85 const std::vector<PartialTensorShape>& output_shapes)
93 output_shapes_(output_shapes) {
108 const std::vector<PartialTensorShape>& output_shapes() const override {
81 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& func, std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, int64 block_length, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dscan_dataset_op.cc40 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
79 const std::vector<PartialTensorShape>& output_shapes)
87 output_shapes_(output_shapes) {
102 const std::vector<PartialTensorShape>& output_shapes() const override {
200 if (!output_shapes()[output_index].IsCompatibleWith(
204 " (expected ", output_shapes()[output_index].DebugString(),
74 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& func, std::vector<Tensor> initial_state, std::unique_ptr<CapturedFunction> captured_func, const DataTypeVector& state_types, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dgroup_by_window_dataset_op.cc39 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
105 const std::vector<PartialTensorShape>& output_shapes)
115 output_shapes_(output_shapes) {
130 const std::vector<PartialTensorShape>& output_shapes() const override {
452 dataset()->input_->output_shapes(), &group_dataset));
98 Dataset(OpKernelContext* ctx, const DatasetBase* input, const NameAttrList& key_func, const NameAttrList& reduce_func, const NameAttrList& window_size_func, std::unique_ptr<CapturedFunction> captured_key_func, std::unique_ptr<CapturedFunction> captured_reduce_func, std::unique_ptr<CapturedFunction> captured_window_size_func, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes) argument
H A Dmap_and_batch_dataset_op.cc42 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
83 const std::vector<PartialTensorShape>& output_shapes,
90 output_shapes_(output_shapes),
108 const std::vector<PartialTensorShape>& output_shapes() const override {
81 Dataset(const DatasetBase* input, int64 batch_size, int64 num_parallel_batches, const DataTypeVector& output_types, const std::vector<PartialTensorShape>& output_shapes, std::unique_ptr<CapturedFunction> captured_func, const Eigen::ThreadPoolDevice* device) argument
/external/tensorflow/tensorflow/python/data/ops/
H A Dreaders.py68 def output_shapes(self): member in class:TextLineDataset
113 def output_shapes(self): member in class:TFRecordDataset
167 def output_shapes(self): member in class:FixedLengthRecordDataset
H A Diterator_ops.py57 output_shapes, output_classes):
71 output_shapes: A nested structure of `tf.TensorShape` objects
81 self._output_shapes = output_shapes
88 output_shapes=None,
139 output_shapes: (Optional.) A nested structure of `tf.TensorShape` objects
153 TypeError: If the structures of `output_shapes` and `output_types` are
157 if output_shapes is None:
158 output_shapes = nest.map_structure(
161 output_shapes = nest.map_structure_up_to(
162 output_types, tensor_shape.as_shape, output_shapes)
397 def output_shapes(self): member in class:Iterator
[all...]
/external/tensorflow/tensorflow/contrib/training/python/training/
H A Dtensor_queue_dataset.py51 dataset_ops._partial_shape_to_tensor, input_dataset.output_shapes)
54 input_dataset.output_shapes, dataset_ops._partial_shape_to_tensor,
60 input_dataset.output_shapes, dataset_ops._padding_value_to_tensor,
74 output_shapes=nest.flatten(
75 sparse.as_dense_shapes(self.output_shapes, self.output_classes)))
87 def output_shapes(self): member in class:_PrependFromQueueAndPaddedBatchDataset
150 `output_types`. If not provided, the incoming dataset's `output_shapes`
174 The components' dtypes and shapes must be compatible with the `output_shapes`
191 `dataset.output_types[1]` and `dataset.output_shapes[1]` (the non-queue
/external/tensorflow/tensorflow/contrib/eager/python/
H A Ddatasets.py81 self._output_shapes = dataset.output_shapes
90 output_shapes=self._flat_output_shapes)
153 output_shapes=self._flat_output_shapes)
180 def output_shapes(self): member in class:Iterator

Completed in 865 milliseconds

12