/external/tensorflow/tensorflow/core/ops/ |
H A D | dataset_ops.cc | 39 .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 D | unique.py | 67 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 D | error_ops.py | 67 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 D | random_ops.py | 52 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 D | readers.py | 194 nest.flatten(self.output_shapes)) 201 def output_shapes(self): member in class:SqlDataset
|
H A D | scan_ops.py | 78 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 D | shuffle_ops.py | 71 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 D | grouping.py | 46 (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 D | interleave_ops.py | 46 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 D | stats_ops.py | 167 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 D | batching.py | 140 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 D | dataset_ops.py | 55 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 D | kafka_dataset_ops.py | 69 def output_shapes(self): member in class:KafkaDataset
|
/external/tensorflow/tensorflow/core/kernels/data/ |
H A D | window_dataset.cc | 24 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 D | sql_dataset_ops.cc | 36 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 D | map_dataset_op.cc | 36 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 D | flat_map_dataset_op.cc | 37 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 D | interleave_dataset_op.cc | 38 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 D | scan_dataset_op.cc | 40 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 D | group_by_window_dataset_op.cc | 39 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 D | map_and_batch_dataset_op.cc | 42 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 D | readers.py | 68 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 D | iterator_ops.py | 57 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 D | tensor_queue_dataset.py | 51 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 D | datasets.py | 81 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
|