| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | shuffle_and_repeat_fusion_test.cc | 28 constexpr char kOutputShapes[] = "output_shapes"; variable
38 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
39 common_attrs[0] = std::make_pair(kOutputShapes, shapes_attr); in TEST()
94 {kOutputShapes, kOutputTypes, kReshuffleEachIteration}) { in TEST()
106 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
107 common_attrs[0] = std::make_pair(kOutputShapes, shapes_attr); in TEST()
158 for (const auto &attr : {kOutputShapes, kOutputTypes}) { in TEST()
171 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
172 common_attrs[0] = std::make_pair(kOutputShapes, shapes_attr); in TEST()
231 {kOutputShapes, kOutputTypes, kReshuffleEachIteration}) { in TEST()
[all …]
|
| D | shuffle_and_repeat_fusion.cc | 42 constexpr char kOutputShapes[] = "output_shapes"; variable
71 for (auto key : {kOutputShapes, kOutputTypes, kReshuffleEachIteration}) { in FuseShuffleV1AndRepeat()
107 for (auto key : {kOutputShapes, kOutputTypes}) { in FuseShuffleV2AndRepeat()
145 for (auto key : {kOutputShapes, kOutputTypes, kReshuffleEachIteration}) { in FuseShuffleV3AndRepeat()
|
| D | auto_shard.cc | 57 constexpr char kOutputShapes[] = "output_shapes"; variable
167 if (add_after->attr().count(kOutputShapes) > 0) { in AddShardNode()
168 graph_utils::CopyAttribute(kOutputShapes, *add_after, &new_node); in AddShardNode()
171 (*(new_node.mutable_attr()))[kOutputShapes] in AddShardNode()
220 graph_utils::CopyAttribute(kOutputShapes, *add_after, &new_node); in AddShuffleDataset()
247 graph_utils::CopyAttribute(kOutputShapes, *add_after, &new_node); in AddShuffleDatasetV2()
274 graph_utils::CopyAttribute(kOutputShapes, *add_after, &new_node); in AddShuffleDatasetV3()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | lmdb_dataset_op.h | 29 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | random_dataset_op.h | 35 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | directed_interleave_dataset_op.h | 31 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | unique_dataset_op.h | 29 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | assert_cardinality_dataset_op.cc | 37 AssertCardinalityDatasetOp::kOutputShapes; member in tensorflow::data::experimental::AssertCardinalityDatasetOp
172 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AssertCardinalityDatasetOp()
|
| D | assert_next_dataset_op.cc | 32 /* static */ constexpr const char* const AssertNextDatasetOp::kOutputShapes; member in tensorflow::data::experimental::AssertNextDatasetOp
155 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in AssertNextDatasetOp()
|
| D | sampling_dataset_op.h | 37 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | compression_ops.h | 34 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | assert_cardinality_dataset_op.h | 30 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | assert_next_dataset_op.h | 30 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | zip_dataset_op.h | 28 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | repeat_dataset_op.h | 29 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | skip_dataset_op.h | 29 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | tensor_dataset_op.h | 28 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | concatenate_dataset_op.h | 29 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | tensor_slice_dataset_op.h | 28 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | parallel_batch_dataset_op.h | 32 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | range_dataset_op.h | 30 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | tensor_dataset_op.cc | 32 /* static */ constexpr const char* const TensorDatasetOp::kOutputShapes; member in tensorflow::data::TensorDatasetOp
152 OP_REQUIRES_OK(ctx, ctx->GetAttr(kOutputShapes, &output_shapes_)); in TensorDatasetOp()
|
| D | batch_dataset_op.h | 31 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | shard_dataset_op.h | 31 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|
| D | window_dataset_op.h | 37 static constexpr const char* const kOutputShapes = "output_shapes"; variable
|