Searched defs:num_replicas_ (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ | ||
| D | auto_shard.h | 48 int64_t num_replicas_; variable |
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ | ||
| D | auto_shard_dataset_op.h | 46 int64_t num_replicas_; variable |
| D | auto_shard_dataset_op_test.cc | 78 int64_t num_replicas_; member in tensorflow::data::experimental::__anon6457e29b0111::AutoShardDatasetParams |
| D | rebatch_dataset_op.cc | 266 const int64_t num_replicas_; member in tensorflow::data::experimental::__anon1d26b22b0111::RebatchDatasetOp::Dataset |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ | ||
| D | auto_parallel.h | 48 int num_replicas_; variable |
| /external/tensorflow/tensorflow/compiler/xla/client/ | ||
| D | executable_build_options.h | 181 int num_replicas_ = 1; variable |
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ | ||
| D | spmd_partitioner.h | 286 const int64_t num_replicas_; variable |
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | tfrt_cpu_pjrt_client.h | 562 int num_replicas_; variable |