| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 114 def num_replicas(self): member in DeviceAssignment
172 num_replicas=1): argument
180 num_replicas=1): argument
|
| D | tpu_context.py | 105 def num_replicas(self): member in TPUContext
361 def num_replicas(self): member in _InternalTPUContext
|
| D | tpu.py | 139 def __init__(self, name, num_replicas, pivot): argument
|
| D | tensor_tracer.py | 1540 num_replicas=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.h | 72 int num_replicas() const { return num_replicas_; } in num_replicas() function
|
| D | executable_build_options.cc | 62 int num_replicas) { in set_num_replicas()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_parallel.h | 29 AutoParallel(int num_replicas) : num_replicas_(num_replicas) { in AutoParallel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 50 int64 num_replicas = 1; member
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.h | 130 int num_replicas() const { in num_replicas() function
|
| D | local_computation_builder.i | 245 int64 num_replicas; variable
|
| D | local_computation_builder.cc | 218 int num_replicas = device_assignment_.replica_count(); in DeviceOrdinals() local
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/ |
| D | resnet50_test.py | 53 def compute_gradients(model, images, labels, num_replicas=1): argument
201 num_replicas=1): argument
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 32 int32 num_replicas = 1; field
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | keras_tpu_variables.py | 261 def replicated_scope(num_replicas): argument
302 def replicated_variable_for_optimizer(num_replicas): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | rewriter_config.proto | 15 int32 num_replicas = 2; field
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 88 int num_replicas = config.num_replicas() ? config.num_replicas() : 1; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 210 int64 num_replicas, bool use_threads) { in ExecuteReplicated()
222 int64 num_replicas, DeviceAssignment* device_assignment, in ExecuteReplicated()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 301 int32 num_replicas = 6; field
|