| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 54 int replica_id, int total_replica_count, in GetParticipatingReplicas()
85 for (int replica_id = 0; replica_id < total_replica_count; ++replica_id) { in GetParticipatingDevices() local
95 int replica_id = logical_ids.first; in GetParticipatingDevices() local
102 for (int replica_id : participating_replicas) { in GetParticipatingDevices() local
|
| D | ar_crs_combiner.cc | 109 int64 replica_id = group.replica_ids(0) / num_partitions; in HasCombinableReplicaGroup() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 143 int replica_id; member
166 int replica_id; member
357 int replica_id = replica_ids_to_copy_to[rank]; in RunCollectiveOp() local
367 int replica_id = replica_ids_to_copy_to[i]; in RunCollectiveOp() local
624 xla::int32 replica_id = in __xla_cpu_runtime_AllToAll() local
712 xla::int32 replica_id = in __xla_cpu_runtime_ReplicaId() local
728 xla::int32 replica_id = in __xla_cpu_runtime_CollectivePermute() local
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_utils.py | 130 def select_replica(replica_id, structured): argument
146 def select_replica_mirrored(replica_id, structured): argument
|
| D | mirrored_run.py | 242 def __init__(self, dist, coord, replica_id, devices, variable_creator_fn, argument
|
| D | collective_all_reduce_strategy.py | 538 replica_id, argument
930 def _get_replica_id_in_sync_group(self, replica_id): argument
|
| D | mirrored_strategy.py | 469 replica_id, argument
860 def _get_replica_id_in_sync_group(self, replica_id): argument
|
| D | cross_device_ops.py | 532 def _all_reduce(self, reduce_op, value, replica_id, options): argument
1081 def _all_reduce(self, reduce_op, value, replica_id, options): argument
|
| D | parameter_server_strategy.py | 704 def _get_replica_id_in_sync_group(self, replica_id): argument
|
| D | tpu_strategy.py | 1447 def replicated_fn(replica_id, replica_args, replica_kwargs): argument
|
| D | distribute_lib.py | 254 def __init__(self, replica_id): argument
3531 def _get_replica_id_in_sync_group(self, replica_id): argument
|
| D | values.py | 679 def _get_replica(self, replica_id): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_to_island.cc | 69 unsigned replica_id) { in GetDeviceOrdinal()
92 OpBuilder& builder, Location loc, Region& region, int replica_id, in UpdateRegionReplicateVariantOps()
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | steps_db.proto | 56 uint32 replica_id = 8; field
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.cc | 96 Status Client::TransferToInfeed(const LiteralSlice& literal, int64 replica_id, in TransferToInfeed()
119 const Shape* shape_with_layout, int64 replica_id, in TransferFromOutfeed()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 465 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveHeadOutsideCompilationToHost() local
554 for (int replica_id = 0; replica_id < num_replicas; ++replica_id) { in MoveHeadOutsideCompilationToHost() local
618 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveHeadOutsideCompilationToHost() local
1065 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveTailOutsideCompilationToHost() local
1097 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveTailOutsideCompilationToHost() local
1116 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveTailOutsideCompilationToHost() local
1174 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in MoveTailOutsideCompilationToHost() local
1525 for (int replica_id = 0; replica_id < num_replicas; replica_id++) { in RemoveEdgesBetweenArgAndRetval() local
|
| D | distributed_tpu_rewrite_pass.cc | 385 int replica_id; in FindTaggedNodes() local
642 int replica_id; member
793 const PartialTensorShape& partial_tensor_shape, int replica_id, in CreateOrGetSplitNodesForInputSharding()
1021 const PartialTensorShape& inferred_shape, int replica_id, in CreateConcatNodesForRetval()
3747 int replica_id; in LowerOutsideCompilationFunctionalNodes() local
3848 int replica_id; in LowerOutsideCompilationFunctionalNodes() local
4182 int replica_id; in UpdateHeadTailOutsideCompilation() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1299 def _inspect_summary_cache(self, cache, replica_id, step_num, output_stream, argument
1381 def _flush_fun(cache, replica_id, step_num): argument
1911 def write_if_core_0(step, replica_id, tt_summary): argument
|
| D | tpu_embedding_v2_correctness_test.py | 435 def _get_replica_numpy(self, structured, strategy, replica_id): argument
|
| D | tpu_embedding_v2_test.py | 1118 def _get_replica_numpy(self, structured, strategy, replica_id): argument
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla.proto | 417 int64 replica_id = 2; field
428 int64 replica_id = 2; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 92 int64 replica_id; member
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt.proto | 116 int32 replica_id = 1; field
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops_test.py | 1878 def __init__(self, graph, replica_id): argument
1935 def __init__(self, graph, replica_id, op_to_colocate_with): argument
2000 def __init__(self, graph, replica_id, dependency_op): argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 393 replica_id = gen_xla_ops.xla_replica_id variable
|