Home
last modified time | relevance | path

Searched defs:num_replicas_ (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dauto_shard.h48 int64_t num_replicas_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.h286 const int64_t num_replicas_; variable
600 int64_t num_replicas_; variable
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op.h46 int64_t num_replicas_; variable
H A Dauto_shard_dataset_op_test.cc78 int64_t num_replicas_; member in tensorflow::data::experimental::__anon268204ac0111::AutoShardDatasetParams
H A Drebatch_dataset_op.cc266 const int64_t num_replicas_; member in tensorflow::data::experimental::__anondafc533c0111::RebatchDatasetOp::Dataset
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_parallel.h48 int num_replicas_; variable
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/data/samplers/
H A Ddistributed.h61 size_t num_replicas_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h181 int num_replicas_ = 1; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.h562 int num_replicas_; variable