Home
last modified time | relevance | path

Searched defs:replica (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.cc94 for (int replica = 0; replica < replica_count(); ++replica) { in Serialize() local
117 for (int replica = 0; replica < proto.replica_count(); ++replica) { in Deserialize() local
130 for (int replica = 0; replica < replica_count(); ++replica) { in ToString() local
138 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
150 for (int replica = 0; replica < replica_count; ++replica) { in AssignDevices() local
Dtpu_computation_placer.cc37 StatusOr<int> TpuComputationPlacer::DeviceId(int replica, int computation, in DeviceId()
Dservice.cc252 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in ResolveAndValidateArguments() local
396 for (int64_t replica = 0, end = replicas.size(); replica < end; ++replica) { in ExecuteParallelAndRegisterResult() local
408 for (int64_t replica = 0; replica < n; ++replica) { in ExecuteParallelAndRegisterResult() local
517 for (int64_t replica = 0; replica < replicas.size(); ++replica) { in ExecuteAndRegisterResult() local
1229 for (int replica = 0; replica < options_.number_of_replicas(); ++replica) { in Replicas() local
Dall_reduce_folder.cc117 for (int64_t replica = 0; replica < num_replicas; ++replica) { in FoldReplicaGroups() local
Dhlo_runner.cc385 [&](int64_t replica) { return options.arguments.size(); }, in ExecuteReplicated()
386 [&](int64_t replica, int64_t index) { return options.arguments[index]; }, in ExecuteReplicated()
Dconditional_code_motion.cc377 bool replica = true; in FindSpecialConverts() local
/external/tensorflow/tensorflow/python/framework/
Ddevice_spec.py109 replica=None, argument
271 def replica(self): member in DeviceSpecV2
370 def _components_to_string(job, replica, task, device_type, device_index): argument
426 def replica(self, replica): member in DeviceSpecV1
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreplicate_to_island.cc163 auto replica = builder.create<tf_executor::IslandOp>( in ExpandReplicateIntoReplicas() local
267 for (auto& replica : replicas) island_operands.push_back(replica.control()); in CreateIslandsFromReplicate() local
283 for (auto& replica : replicas) in CreateIslandsFromReplicate() local
Dtpu_rewrite_pass.cc415 for (int replica = 0; replica < num_replicas; ++replica) in AssignDevicesToReplicate() local
427 for (int replica = 0; replica < num_replicas; ++replica) in AssignDevicesToReplicate() local
/external/guava/guava-tests/test/com/google/common/collect/
DMinMaxPriorityQueueTest.java202 SortedMap<Integer, AtomicInteger> replica = Maps.newTreeMap(); in testHeapIntact() local
897 private static void insertIntoReplica(Map<Integer, AtomicInteger> replica, int newValue) { in insertIntoReplica()
906 SortedMap<Integer, AtomicInteger> replica, int minValue) { in removeMinFromReplica()
913 SortedMap<Integer, AtomicInteger> replica, int maxValue) { in removeMaxFromReplica()
919 private static void removeFromReplica(Map<Integer, AtomicInteger> replica, int value) { in removeFromReplica()
/external/guava/android/guava-tests/test/com/google/common/collect/
DMinMaxPriorityQueueTest.java202 SortedMap<Integer, AtomicInteger> replica = Maps.newTreeMap(); in testHeapIntact() local
897 private static void insertIntoReplica(Map<Integer, AtomicInteger> replica, int newValue) { in insertIntoReplica()
906 SortedMap<Integer, AtomicInteger> replica, int minValue) { in removeMinFromReplica()
913 SortedMap<Integer, AtomicInteger> replica, int maxValue) { in removeMaxFromReplica()
919 private static void removeFromReplica(Map<Integer, AtomicInteger> replica, int value) { in removeFromReplica()
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc79 static string DeviceName(const string& job, int replica, int task, in DeviceName()
92 string DeviceNameUtils::FullName(const string& job, int replica, int task, in FullName()
98 string LegacyName(const string& job, int replica, int task, const string& type, in LegacyName()
Ddevice_name_utils.h84 int replica = 0; member
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc151 for (int replica = 0; replica < num_replicas; ++replica) { in GetDefaultDeviceAssignment() local
532 for (int replica = 0; replica < num_replicas; ++replica) { in PyTpuExecutable() local
557 absl::Span<PyTpuBuffer* const> this_core_arguments, int replica, in ExecuteHelper()
724 const int replica = local_logical_device_ids_[i].first; in ExecuteOnLocalDevices() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dcross_replica_ops.cc43 int64_t replica = group_assignment[num_replica_per_group * g + i]; in Convert() local
/external/tensorflow/tensorflow/python/distribute/
Ddistribution_strategy_context.py39 def __init__(self, dist, cross, replica): argument
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc166 for (int replica = 0; replica < devices.size(); ++replica) { in DevicesToDeviceAssignment() local
1900 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in EnqueueExecution()
2092 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
2226 const int replica = addressable_device_logical_ids_[0].replica; in Execute() local
2237 const int replica = addressable_device_logical_ids_[i].replica; in Execute() local
2293 const int replica = addressable_device_logical_ids_[i].replica; in Execute() local
2418 for (int replica = 0; replica < num_replicas; ++replica) { in GetExecutableExtras() local
Dtfrt_cpu_pjrt_client.cc358 for (int replica = 0; replica < num_replicas; ++replica) { in Compile() local
1304 absl::Span<PjRtBuffer* const> argument_handles, int replica, int partition, in ExecuteHelper()
1651 const int replica = addressable_device_logical_ids_[0].replica; in Execute() local
1682 const int replica = addressable_device_logical_ids_[i].replica; in Execute() local
Dpjrt_client.h1120 int replica; member
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_helpers.cc123 int replica = (*host_system_devices)[0]->parsed_name().replica; in GetHostSystemDevices() local
Ddistributed_tpu_rewrite_pass.cc1055 for (int replica = 0; replica < num_replicas; ++replica) { in CreateOrGetXlaSplitNodeForDistributedArg() local
1115 for (int replica = 0; replica < num_replicas; ++replica) { in CreateOrGetXlaSplitNodeForVariableArg() local
1680 for (int replica = 0; replica < num_replicas; ++replica) { in ParseDeviceAssignmentAttr() local
1747 for (int replica = 0; replica < num_replicas; ++replica) { in BuildGeneralDeviceAssignment() local
3100 for (int replica = 1; replica < num_replicas; ++replica) { in CreatePartitionedDummyVarArgs() local
3488 for (int64_t replica = 0; replica < params_info.NumReplicas(); ++replica) { in BuildExecuteNodes() local
Dencapsulate_tpu_computations_pass.cc2667 for (int replica = 0; replica < num_replicas; ++replica) { in BuildTPUReplicateOps() local
2754 for (int replica = 0; replica < num_replicas; ++replica) { in BuildTPUReplicateOps() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc94 auto replica = system_devices[0].replica; in GetTPUSystemDevices() local
381 for (int replica = 0; replica < num_replicas; ++replica) { in GetGeneralTPUExecutionDeviceAssignment() local
/external/skia/tools/fm/
Dfm.cpp443 for (int replica = 0; replica < replicas; replica++) { in main() local
574 SkTaskGroup{}.batch(replicas, [=](int replica) { in main()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc638 BlockArgument block_arg, unsigned replica) { in GetReplicaOperandIndexForBlockArgument()
649 unsigned replica) { in GetReplicaOperandForBlockArgument()