Home
last modified time | relevance | path

Searched refs:output_shapes_ (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc35 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ChooseFastestDatasetOp()
69 ctx, inputs[i]->output_shapes().size() == output_shapes_.size(), in MakeDataset()
74 " components. Expected to have ", output_shapes_.size(), in MakeDataset()
76 for (size_t j = 0, num_components = output_shapes_.size(); in MakeDataset()
80 output_shapes_[j] in MakeDataset()
89 output_shapes_[j], ".")); in MakeDataset()
90 output_shapes_[j] = std::move(result); in MakeDataset()
111 *output = new Dataset(ctx, std::move(inputs), output_types_, output_shapes_, in MakeDataset()
125 output_shapes_(output_shapes), in Dataset()
151 return output_shapes_; in output_shapes()
[all …]
Dnon_serializable_dataset_op.cc30 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in NonSerializableDatasetOp()
36 *output = new Dataset(ctx, input, output_types_, output_shapes_); in MakeDataset()
48 output_shapes_(output_shapes) { in Dataset()
64 return output_shapes_; in output_shapes()
119 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon4d42c3f70111::NonSerializableDatasetOp::Dataset
123 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon4d42c3f70111::NonSerializableDatasetOp
Dassert_next_dataset_op.cc32 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in AssertNextDatasetOp()
42 new Dataset(ctx, input, transformations, output_types_, output_shapes_); in MakeDataset()
56 output_shapes_(output_shapes) { in Dataset()
72 return output_shapes_; in output_shapes()
151 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon9dfb5fa60111::AssertNextDatasetOp::Dataset
155 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon9dfb5fa60111::AssertNextDatasetOp
Dsql_dataset_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in SqlDatasetOp()
48 for (const PartialTensorShape& pts : output_shapes_) { in SqlDatasetOp()
75 output_types_, output_shapes_); in MakeDataset()
90 output_shapes_(output_shapes) {} in Dataset()
103 return output_shapes_; in output_shapes()
211 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon6e0d6b260111::SqlDatasetOp::Dataset
214 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon6e0d6b260111::SqlDatasetOp
Dchoose_fastest_branch_dataset_op.cc43 output_shapes_(output_shapes), in WrapperDataset()
51 return *output_shapes_; in output_shapes()
119 const std::vector<PartialTensorShape>* const output_shapes_; member in tensorflow::data::__anonf4b3f19d0111::WrapperDataset
136 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ChooseFastestBranchDatasetOp()
182 output_types_, output_shapes_, num_elements_per_branch_, in MakeDataset()
201 output_shapes_(output_shapes), in Dataset()
222 return output_shapes_; in output_shapes()
487 &dataset()->output_shapes_, input_impl_.get()); in MakeCurrentIterator()
536 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonf4b3f19d0111::ChooseFastestBranchDatasetOp::Dataset
546 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonf4b3f19d0111::ChooseFastestBranchDatasetOp
Ddirected_interleave_dataset_op.cc75 output_shapes_ = data_inputs_[0]->output_shapes(); in Dataset()
80 for (size_t j = 0; j < output_shapes_.size(); ++j) { in Dataset()
81 output_shapes_[j] = MostSpecificCompatibleShape( in Dataset()
82 output_shapes_[j], data_input->output_shapes()[j]); in Dataset()
105 return output_shapes_; in output_shapes()
276 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon9a3736c80111::DirectedInterleaveDatasetOp::Dataset
Drebatch_dataset_op.cc30 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in RebatchDatasetOp()
43 new Dataset(ctx, input, num_workers, output_types_, output_shapes_); in MakeDataset()
93 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonc56d986f0111::RebatchDatasetOp
Dindexed_dataset_op.cc106 : output_dtypes_(output_dtypes), output_shapes_(output_shapes) {} in MaterializedDatasetResource()
129 return output_shapes_; in output_shapes()
138 VerifyShapesCompatible(output_shapes_, dataset->output_shapes())); in set_materialized_dataset()
147 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon0a1add170111::MaterializedDatasetResource
254 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in MaterializedHandleOp()
286 output_dtypes_, output_shapes_); in Compute()
314 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
322 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon0a1add170111::MaterializedHandleOp
Dauto_shard_dataset_op.cc30 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in AutoShardDatasetOp()
49 output_types_, output_shapes_); in MakeDataset()
110 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonb61025000111::AutoShardDatasetOp
Dscan_dataset_op.cc39 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ScanDatasetOp()
58 output_shapes_, preserve_cardinality_); in MakeDataset()
78 output_shapes_(output_shapes), in Dataset()
95 return output_shapes_; in output_shapes()
291 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anona5deed3b0111::ScanDatasetOp::Dataset
297 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anona5deed3b0111::ScanDatasetOp
/external/tensorflow/tensorflow/core/kernels/data/
Dfilter_by_component_dataset_op.cc38 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in FilterByLastComponentDatasetOp()
43 *output = new Dataset(ctx, input, output_types_, output_shapes_); in MakeDataset()
49 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anond0e1a64b0111::FilterByLastComponentDatasetOp
59 output_shapes_(std::move(output_shapes)) { in Dataset()
75 return output_shapes_; in output_shapes()
98 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anond0e1a64b0111::FilterByLastComponentDatasetOp::Dataset
Dmulti_device_iterator_ops.cc55 output_shapes_(output_shapes), in MultiDeviceIterator()
75 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in Init()
111 return output_shapes_; in output_shapes()
355 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator
377 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in MultiDeviceIteratorHandleOp()
422 context->env(), output_types_, output_shapes_, devices_, in Compute()
440 output_shapes_, devices_, in Compute()
470 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
478 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon2db6daca0111::MultiDeviceIteratorHandleOp
608 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in MultiDeviceIteratorFromStringHandleOp()
[all …]
Diterator_ops.cc68 output_shapes_(output_shapes) {} in IteratorResource()
177 output_shapes_, new_state->iterator->output_shapes())); in Restore()
249 VerifyShapesCompatible(output_shapes_, iterator->output_shapes())); in SetIteratorFromDataset()
262 return output_shapes_; in output_shapes()
300 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::IteratorResource
396 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in IteratorHandleOp()
447 context->env(), output_dtypes_, output_shapes_, in Compute()
472 VerifyShapesCompatible(output_shapes_, resource->output_shapes())); in VerifyResource()
505 OP_REQUIRES_OK(context, context->GetAttr("output_shapes", &output_shapes_)); in AnonymousIteratorHandleOp()
534 context->env(), output_dtypes_, output_shapes_, graph_def_version_, in Compute()
[all …]
Diterator_ops.h73 std::vector<PartialTensorShape> output_shapes_; variable
97 std::vector<PartialTensorShape> output_shapes_; variable
127 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in IteratorGetNextAsOptionalOp()
135 std::vector<PartialTensorShape> output_shapes_; variable
161 std::vector<PartialTensorShape> output_shapes_; variable
Dgenerator_dataset_op.cc43 output_shapes_(output_shapes) {} in Dataset()
54 return output_shapes_; in output_shapes()
152 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::GeneratorDatasetOp::Dataset
161 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in GeneratorDatasetOp()
181 std::move(finalize_func), output_types_, output_shapes_); in MakeDataset()
Dflat_map_dataset_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in FlatMapDatasetOp()
46 output_types_, output_shapes_); in MakeDataset()
62 output_shapes_(output_shapes) { in Dataset()
79 return output_shapes_; in output_shapes()
268 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anond045031d0111::FlatMapDatasetOp::Dataset
273 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anond045031d0111::FlatMapDatasetOp
Dgraph_rewrite_dataset.h43 output_shapes_(output_shapes) { in GraphRewriteDataset()
59 return output_shapes_; in output_shapes()
89 const std::vector<PartialTensorShape> output_shapes_; variable
Dparallel_map_dataset_op.cc43 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in ParallelMapDatasetOp()
74 output_shapes_, use_inter_op_parallelism_, sloppy_, in MakeDataset()
94 output_shapes_(output_shapes), in Dataset()
127 return output_shapes_; in output_shapes()
277 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon58e4a4830111::ParallelMapDatasetOp::Dataset
287 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon58e4a4830111::ParallelMapDatasetOp
Dmap_dataset_op.cc39 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in MapDatasetOp()
91 output_shapes_, use_inter_op_parallelism_, in MakeDataset()
112 output_shapes_(output_shapes), in Dataset()
129 return output_shapes_; in output_shapes()
269 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonb3c9acd70111::MapDatasetOp::Dataset
274 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonb3c9acd70111::MapDatasetOp
Doptimize_dataset_op.cc38 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in OptimizeDatasetOp()
48 new Dataset(ctx, input, optimizations, output_types_, output_shapes_); in MakeDataset()
92 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon90faeaaa0111::OptimizeDatasetOp
Dbatch_dataset_op.cc66 output_shapes_.reserve(input_shapes.size()); in Dataset()
69 output_shapes_.emplace_back( in Dataset()
72 output_shapes_.emplace_back( in Dataset()
91 return output_shapes_; in output_shapes()
251 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anonb889a47b0111::BatchDatasetOp::Dataset
Dmap_defun_op.cc55 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in MapDefunOp()
63 OP_REQUIRES(ctx, ctx->num_outputs() == output_shapes_.size(), in MapDefunOp()
297 output_shapes_, max_intra_op_parallelism_); in SetupArgs()
306 if (output_shapes_.at(i).IsFullyDefined()) { in SetupOutputs()
309 output_shapes_.at(i).AsTensorShape(&output_shape); in SetupOutputs()
318 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anone81855e30111::MapDefunOp
Doptional_ops.h179 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in OptionalGetValueOp()
182 ctx, output_shapes_.size() == output_types_.size(), in OptionalGetValueOp()
186 "output_shapes: ", output_shapes_.size())); in OptionalGetValueOp()
193 std::vector<PartialTensorShape> output_shapes_; variable
Dzip_dataset_op.cc51 output_shapes_.insert(output_shapes_.end(), in Dataset()
74 return output_shapes_; in output_shapes()
194 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon8a8355ec0111::ZipDatasetOp::Dataset
Dinterleave_dataset_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_)); in InterleaveDatasetOp()
66 block_length, output_types_, output_shapes_); in MakeDataset()
84 output_shapes_(output_shapes) { in Dataset()
100 return output_shapes_; in output_shapes()
325 const std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon811adb680111::InterleaveDatasetOp::Dataset
330 std::vector<PartialTensorShape> output_shapes_; member in tensorflow::data::__anon811adb680111::InterleaveDatasetOp

12