Searched defs:num_replicas_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ | ||
| D | auto_shard_dataset_op.h | 46 int64 num_replicas_; variable |
| D | auto_shard_dataset_op_test.cc | 73 int64 num_replicas_; member in tensorflow::data::experimental::__anon8ac557bd0111::AutoShardDatasetParams |
| D | rebatch_dataset_op.cc | 260 const int64 num_replicas_; member in tensorflow::data::experimental::__anon714a254d0111::RebatchDatasetOp::Dataset |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ | ||
| D | auto_parallel.h | 51 int num_replicas_; variable |
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ | ||
| D | auto_shard.h | 51 int64 num_replicas_; variable |
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | ar_crs_combiner.h | 172 int num_replicas_; variable |
| /external/tensorflow/tensorflow/compiler/xla/client/ | ||
| D | executable_build_options.h | 147 int num_replicas_ = 1; variable |
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | spmd_partitioner.h | 247 const int64 num_replicas_; variable |