| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | non_serializable_dataset_op.cc | 31 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in NonSerializableDatasetOp()
37 *output = new Dataset(ctx, input, output_types_, output_shapes_); in MakeDataset()
49 output_shapes_(output_shapes) { in Dataset()
65 return output_shapes_; in output_shapes()
135 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon650a91330111::NonSerializableDatasetOp::Dataset
139 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon650a91330111::NonSerializableDatasetOp
|
| D | choose_fastest_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ChooseFastestDatasetOp()
70 ctx, inputs[i]->output_shapes().size() == output_shapes_.size(), in MakeDataset()
75 " components. Expected to have ", output_shapes_.size(), in MakeDataset()
77 for (size_t j = 0, num_components = output_shapes_.size(); in MakeDataset()
81 output_shapes_[j] in MakeDataset()
90 output_shapes_[j], ".")); in MakeDataset()
91 output_shapes_[j] = std::move(result); in MakeDataset()
112 *output = new Dataset(ctx, std::move(inputs), output_types_, output_shapes_, in MakeDataset()
126 output_shapes_(output_shapes), in Dataset()
152 return output_shapes_; in output_shapes()
[all …]
|
| D | list_dataset_op.cc | 55 output_shapes_(output_shapes) {} in Dataset()
73 return output_shapes_; in output_shapes()
189 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::ListDatasetOp::Dataset
195 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in ListDatasetOp()
203 output_shapes_, output_shapes_.size()); in MakeDataset()
207 ctx, VerifyShapesCompatible((*output)->output_shapes(), output_shapes_)); in MakeDataset()
|
| D | sql_dataset_op.cc | 35 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in SqlDatasetOp()
46 for (const PartialTensorShape& pts : output_shapes_) { in SqlDatasetOp()
73 output_types_, output_shapes_); in MakeDataset()
88 output_shapes_(output_shapes) {} in Dataset()
101 return output_shapes_; in output_shapes()
227 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anondf2706e20111::SqlDatasetOp::Dataset
230 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anondf2706e20111::SqlDatasetOp
|
| D | rebatch_dataset_op.cc | 38 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in RebatchDatasetOp()
51 new Dataset(ctx, input, num_replicas, output_types_, output_shapes_); in MakeDataset()
64 output_shapes_(output_shapes), in Dataset()
85 return output_shapes_; in output_shapes()
268 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon1d26b22b0111::RebatchDatasetOp::Dataset
273 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon1d26b22b0111::RebatchDatasetOp
290 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in RebatchDatasetV2Op()
313 output_types_, output_shapes_); in MakeDataset()
328 output_shapes_(output_shapes), in Dataset()
348 return output_shapes_; in output_shapes()
[all …]
|
| D | assert_next_dataset_op.cc | 44 output_shapes_(output_shapes) { in Dataset()
58 return output_shapes_; in output_shapes()
149 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::AssertNextDatasetOp::Dataset
155 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AssertNextDatasetOp()
164 new Dataset(ctx, input, transformations, output_types_, output_shapes_); in MakeDataset()
|
| D | assert_cardinality_dataset_op.cc | 48 output_shapes_(output_shapes) { in Dataset()
62 return output_shapes_; in output_shapes()
165 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::AssertCardinalityDatasetOp::Dataset
172 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AssertCardinalityDatasetOp()
181 *output = new Dataset(ctx, input, cardinality, output_types_, output_shapes_); in MakeDataset()
|
| D | directed_interleave_dataset_op.cc | 57 output_shapes_ = data_inputs_[0]->output_shapes(); in Dataset()
62 for (size_t j = 0; j < output_shapes_.size(); ++j) { in Dataset()
63 output_shapes_[j] = MostSpecificCompatibleShape( in Dataset()
64 output_shapes_[j], data_input->output_shapes()[j]); in Dataset()
93 return output_shapes_; in output_shapes()
320 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::DirectedInterleaveDatasetOp::Dataset
|
| D | choose_fastest_branch_dataset_op.cc | 45 output_shapes_(output_shapes), in WrapperDataset()
53 return *output_shapes_; in output_shapes()
128 const std::vector<PartialTensorShape>* const output_shapes_; member in tensorflow::data::experimental::__anonbe5964590111::WrapperDataset
152 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ChooseFastestBranchDatasetOp()
197 output_shapes_, num_elements_per_branch_, in MakeDataset()
214 output_shapes_(output_shapes), in Dataset()
235 return output_shapes_; in output_shapes()
570 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anonbe5964590111::ChooseFastestBranchDatasetOp::Dataset
581 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anonbe5964590111::ChooseFastestBranchDatasetOp
|
| D | assert_prev_dataset_op.cc | 143 output_shapes_(output_shapes) { in Dataset()
157 return output_shapes_; in output_shapes()
250 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::AssertPrevDatasetOp::Dataset
256 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AssertPrevDatasetOp()
265 new Dataset(ctx, input, transformations, output_types_, output_shapes_); in MakeDataset()
|
| D | random_access_ops.h | 34 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in GetElementAtIndexOp()
44 std::vector<PartialTensorShape> output_shapes_; variable
|
| D | scan_dataset_op.cc | 46 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ScanDatasetOp()
67 output_shapes_, preserve_cardinality_, use_default_device_); in MakeDataset()
86 output_shapes_(output_shapes), in Dataset()
104 return output_shapes_; in output_shapes()
318 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon3a2c00770111::ScanDatasetOp::Dataset
326 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anon3a2c00770111::ScanDatasetOp
|
| D | group_by_reducer_dataset_op.cc | 47 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in GroupByReducerDatasetOp()
73 output_types_, output_shapes_); in MakeDataset()
93 output_shapes_(output_shapes) { in Dataset()
109 return output_shapes_; in output_shapes()
423 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anona14e20c20111::GroupByReducerDatasetOp::Dataset
431 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anona14e20c20111::GroupByReducerDatasetOp
|
| D | dense_to_sparse_batch_dataset_op.cc | 87 output_shapes_.reserve(1); in Dataset()
90 output_shapes_.push_back(output_shape); in Dataset()
107 return output_shapes_; in output_shapes()
320 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::experimental::__anoncb2181d10111::DenseToSparseBatchDatasetOp::Dataset
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 77 return output_shapes_; in output_shapes()
144 const std::vector<PartialTensorShape> output_shapes_; variable
175 std::vector<PartialTensorShape> output_shapes_; variable
198 std::vector<PartialTensorShape> output_shapes_; variable
246 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in IteratorGetNextOp()
256 std::vector<PartialTensorShape> output_shapes_; variable
273 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in IteratorGetNextAsOptionalOp()
281 std::vector<PartialTensorShape> output_shapes_; variable
300 std::vector<PartialTensorShape> output_shapes_; variable
|
| D | window_dataset.cc | 43 output_shapes_(std::move(output_shapes)) {}
54 return output_shapes_; in output_shapes()
147 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anone003808f0111::Window
154 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in WindowOp()
161 auto element_size = output_shapes_.size(); in MakeDataset()
171 *output = new Window(std::move(elements), output_types_, output_shapes_); in MakeDataset()
176 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anone003808f0111::WindowOp
|
| D | multi_device_iterator_ops.cc | 89 output_shapes_(output_shapes), in MultiDeviceIterator()
114 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in Init()
153 return output_shapes_; in output_shapes()
462 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon2efe57860111::MultiDeviceIterator
485 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in MultiDeviceIteratorHandleOp()
529 context->env(), output_types_, output_shapes_, devices_, in Compute()
547 output_shapes_, devices_, in Compute()
577 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
585 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon2efe57860111::MultiDeviceIteratorHandleOp
607 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AnonymousMultiDeviceIteratorOp()
[all …]
|
| D | optional_ops.h | 73 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in OptionalGetValueOp()
76 ctx, output_shapes_.size() == output_types_.size(), in OptionalGetValueOp()
80 "output_shapes: ", output_shapes_.size())); in OptionalGetValueOp()
87 std::vector<PartialTensorShape> output_shapes_; variable
|
| D | iterator_ops.cc | 103 output_shapes_(output_shapes) { in IteratorResource()
254 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in SetIteratorFromDataset()
440 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in IteratorHandleOp()
492 context->env(), output_dtypes_, output_shapes_, in Compute()
517 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
564 OP_REQUIRES_OK(context, context->GetAttr(kOutputShapes, &output_shapes_)); in AnonymousIteratorHandleOp()
574 *resource = new IteratorResource(ctx->env(), output_dtypes_, output_shapes_, in CreateResource()
629 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ToSingleElementOp()
680 TF_RETURN_IF_ERROR(VerifyShapesCompatible(output_shapes_, components)); in DoCompute()
697 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonbeff1aad0711::ToSingleElementOp
[all …]
|
| D | generator_dataset_op.cc | 60 output_shapes_(output_shapes) {} in Dataset()
71 return output_shapes_; in output_shapes()
192 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::GeneratorDatasetOp::Dataset
205 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in GeneratorDatasetOp()
225 std::move(finalize_func), output_types_, output_shapes_); in MakeDataset()
|
| D | map_dataset_op.cc | 54 output_shapes_(output_shapes) { in Dataset()
69 return output_shapes_; in output_shapes()
233 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::MapDatasetOp::Dataset
247 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in MapDatasetOp()
260 output_shapes_, preserve_cardinality_); in MakeDataset()
|
| D | batch_dataset_op.cc | 75 output_shapes_.reserve(input_shapes.size());
78 output_shapes_.emplace_back(
81 output_shapes_.emplace_back(
102 return output_shapes_; in output_shapes()
283 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::BatchDatasetOp::Dataset
|
| D | map_defun_op.cc | 178 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in MapDefunOp()
186 OP_REQUIRES(ctx, ctx->num_outputs() == output_shapes_.size(), in MapDefunOp()
289 batch_size, output_shapes_, max_intra_op_parallelism_); in SetupArgs()
298 if (output_shapes_.at(i).IsFullyDefined()) { in SetupOutputs()
301 output_shapes_.at(i).AsTensorShape(&output_shape); in SetupOutputs()
|
| D | flat_map_dataset_op.cc | 66 output_shapes_(output_shapes) { in Dataset()
81 return output_shapes_; in output_shapes()
320 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::FlatMapDatasetOp::Dataset
328 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in FlatMapDatasetOp()
338 output_shapes_); in MakeDataset()
|
| D | zip_dataset_op.cc | 52 output_shapes_.insert(output_shapes_.end(), in Dataset()
81 return output_shapes_; in output_shapes()
270 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::ZipDatasetOp::Dataset
|