Searched defs:auto_shard_policy_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ | ||
| D | auto_shard.h | 50 tensorflow::data::AutoShardPolicy auto_shard_policy_; variable |
| /external/tensorflow/tensorflow/core/data/service/ | ||
| D | auto_shard_rewriter.h | 52 const AutoShardPolicy auto_shard_policy_; variable |
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ | ||
| D | auto_shard_dataset_op.h | 45 int64_t auto_shard_policy_; variable |
| D | auto_shard_dataset_op_test.cc | 80 int auto_shard_policy_; member in tensorflow::data::experimental::__anonba308f2c0111::AutoShardDatasetParams |