| /external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | window_dataset.cc | 27 output_shapes_(std::move(output_shapes)) {}
38 return output_shapes_;
83 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26446::WindowDataset
|
| H A D | zip_dataset_op.cc | 51 output_shapes_.insert(output_shapes_.end(),
74 return output_shapes_;
167 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26447::ZipDatasetOp::Dataset
|
| H A D | batch_dataset_op.cc | 55 output_shapes_.reserve(input_shapes.size());
57 output_shapes_.emplace_back(
75 return output_shapes_;
202 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26414::BatchDatasetOp::Dataset
|
| H A D | sql_dataset_ops.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
47 for (const PartialTensorShape& pts : output_shapes_) {
50 "Each element of `output_shapes_` must be a scalar."));
74 output_shapes_);
87 output_shapes_(output_shapes) {}
100 return output_shapes_;
147 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26438::SqlDatasetOp::Dataset
150 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26438::SqlDatasetOp
|
| H A D | concatenate_dataset_op.cc | 55 output_shapes_.push_back(
75 return output_shapes_;
185 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26417::ConcatenateDatasetOp::Dataset
|
| H A D | map_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
54 output_types_, output_shapes_);
70 output_shapes_(output_shapes) {
86 return output_shapes_;
177 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26425::MapDatasetOp::Dataset
182 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26425::MapDatasetOp
|
| H A D | dense_to_sparse_batch_dataset_op.cc | 89 output_shapes_.reserve(1);
92 output_shapes_.push_back(output_shape);
109 return output_shapes_;
291 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26418::DenseToSparseBatchDatasetOp::Dataset
|
| H A D | flat_map_dataset_op.cc | 37 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
55 output_types_, output_shapes_);
71 output_shapes_(output_shapes) {
88 return output_shapes_;
268 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26420::FlatMapDatasetOp::Dataset
273 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26420::FlatMapDatasetOp
|
| H A D | interleave_dataset_op.cc | 38 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
75 block_length, output_types_, output_shapes_);
93 output_shapes_(output_shapes) {
109 return output_shapes_;
313 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26422::InterleaveDatasetOp::Dataset
318 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26422::InterleaveDatasetOp
|
| H A D | padded_batch_dataset_op.cc | 113 output_shapes_.reserve(input_shapes.size());
115 output_shapes_.push_back(
133 return output_shapes_;
343 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26426::PaddedBatchDatasetOp::Dataset
|
| H A D | scan_dataset_op.cc | 40 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
68 output_shapes_);
87 output_shapes_(output_shapes) {
103 return output_shapes_;
263 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26434::ScanDatasetOp::Dataset
269 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26434::ScanDatasetOp
|
| H A D | group_by_window_dataset_op.cc | 39 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
92 std::move(captured_window_size_func), output_types_, output_shapes_);
115 output_shapes_(output_shapes) {
131 return output_shapes_;
520 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26421::GroupByWindowDatasetOp::Dataset
525 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26421::GroupByWindowDatasetOp
|
| H A D | map_and_batch_dataset_op.cc | 42 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
74 output_types_, output_shapes_,
90 output_shapes_(output_shapes),
109 return output_shapes_;
392 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26424::MapAndBatchDatasetOp::Dataset
399 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26424::MapAndBatchDatasetOp
|
| H A D | parallel_interleave_dataset_op.cc | 40 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
89 output_types_, output_shapes_);
108 output_shapes_(output_shapes) {
124 return output_shapes_;
489 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26427::ParallelInterleaveDatasetOp::Dataset
494 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26427::ParallelInterleaveDatasetOp
|
| H A D | parallel_map_dataset_op.cc | 39 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
65 output_shapes_, std::move(captured_func));
81 output_shapes_(output_shapes),
99 return output_shapes_;
388 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26428::ParallelMapDatasetOp::Dataset
394 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26428::ParallelMapDatasetOp
|
| H A D | iterator_ops.cc | 97 output_shapes_(output_shapes) {}
192 VerifyShapesCompatible(output_shapes_, iterator->output_shapes()));
213 return output_shapes_;
229 const std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26423::IteratorResource
441 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
492 output_dtypes_, output_shapes_, graph_def_version_,
523 VerifyShapesCompatible(output_shapes_, resource->output_shapes()));
567 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26423::IteratorHandleOp
670 OP_REQUIRES_OK(ctx, ctx->GetAttr("output_shapes", &output_shapes_));
750 output_dtypes_, output_shapes_, graph_def_version
829 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26423::OneShotIteratorOp
1020 std::vector<PartialTensorShape> output_shapes_; member in class:tensorflow::__anon26423::IteratorFromStringHandleOp
[all...] |
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| H A D | tfprof_node.h | 334 if (!output_shapes_.empty()) {
339 output_shapes_[i] = ShapeProtoToVec(attr.second.list().shape(i));
416 for (const auto& s : output_shapes_) {
470 output_shapes_.clear();
472 auto& shape = output_shapes_[s.first];
702 return output_shapes_;
747 std::map<int, std::vector<int64>> output_shapes_; member in class:tensorflow::tfprof::TFGraphNode
|