/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | choose_fastest_dataset_op.cc | 35 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 …]
|
D | non_serializable_dataset_op.cc | 30 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
|
D | assert_next_dataset_op.cc | 32 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
|
D | sql_dataset_op.cc | 37 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
|
D | choose_fastest_branch_dataset_op.cc | 43 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
|
D | directed_interleave_dataset_op.cc | 75 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
|
D | rebatch_dataset_op.cc | 30 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
|
D | indexed_dataset_op.cc | 106 : 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
|
D | auto_shard_dataset_op.cc | 30 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
|
D | scan_dataset_op.cc | 39 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/ |
D | filter_by_component_dataset_op.cc | 38 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
|
D | multi_device_iterator_ops.cc | 55 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 …]
|
D | iterator_ops.cc | 68 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 …]
|
D | iterator_ops.h | 73 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
|
D | generator_dataset_op.cc | 43 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()
|
D | flat_map_dataset_op.cc | 37 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
|
D | graph_rewrite_dataset.h | 43 output_shapes_(output_shapes) { in GraphRewriteDataset() 59 return output_shapes_; in output_shapes() 89 const std::vector<PartialTensorShape> output_shapes_; variable
|
D | parallel_map_dataset_op.cc | 43 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
|
D | map_dataset_op.cc | 39 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
|
D | optimize_dataset_op.cc | 38 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
|
D | batch_dataset_op.cc | 66 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
|
D | map_defun_op.cc | 55 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
|
D | optional_ops.h | 179 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
|
D | zip_dataset_op.cc | 51 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
|
D | interleave_dataset_op.cc | 37 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
|