| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tpu_computation_placer.cc | 38 int replica_count, in DeviceId()
44 int replica_count, int computation_count) { in AssignDevices()
58 int replica_count, in AssignLocalDevices()
|
| D | computation_placer.h | 44 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment()
50 int replica_count() const { return height(); } in replica_count() function
|
| D | all_reduce_simplifier.h | 31 explicit AllReduceSimplifier(int64 replica_count) in AllReduceSimplifier()
|
| D | computation_placer.cc | 125 int replica_count, in DeviceId()
134 int replica_count, int computation_count) { in AssignDevices()
|
| D | hlo_module_config.h | 126 void set_replica_count(int64 replica_count) { in set_replica_count()
129 int64 replica_count() const { return replica_count_; } in replica_count() function
|
| D | compiler.h | 78 virtual int64 replica_count() const { return 0; } in replica_count() function
|
| D | service.cc | 735 const int64 replica_count = options_.number_of_replicas(); in GetDeviceHandles() local
988 const int64 replica_count = options_.number_of_replicas(); in TransferToInfeed() local
1017 const int64 replica_count = options_.number_of_replicas(); in TransferFromOutfeed() local
|
| D | ar_crs_combiner.cc | 45 int64 replica_count, in ReplaceReplicatedAllReduce()
|
| D | hlo_verifier_test.cc | 882 int64 replica_count = 0; in ReplicaCount() local
891 absl::optional<int64> replica_count = absl::nullopt) { in MakeAllReduceComputation()
|
| D | hlo_verifier.cc | 238 int64 replica_count = hlo->GetModule()->config().replica_count(); in CheckReplicaGroups() local
|
| D | hlo_parser_test.cc | 45 int64 replica_count = 1; member
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_program_group.h | 41 explicit TpuAotCompilationOptions(int64 replica_count) in TpuAotCompilationOptions()
51 int64 replica_count() const override { return replica_count_; } in replica_count() function
|
| D | tpu_compile_op_support.cc | 59 absl::optional<const DeviceAssignment> device_assignment, int replica_count, in CreateModuleConfig()
138 absl::optional<const DeviceAssignment> device_assignment, int replica_count, in CreateModuleConfig()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_gather_thunk.cc | 34 int64 replica_count) { in GetNcclAllGatherConfig()
62 ThunkInfo thunk_info, mlir::lmhlo::AllGatherOp op, int64 replica_count, in NcclAllGatherThunk()
|
| D | nccl_collective_thunk.h | 70 int64 replica_count; member
79 int64 replica_count) { in GetNcclCollectiveConfigForMlir()
|
| D | nccl_all_to_all_thunk.cc | 36 int64 replica_count) { in GetNcclAllToAllConfig()
54 ThunkInfo thunk_info, mlir::lmhlo::AllToAllOp op, int64 replica_count, in NcclAllToAllThunk()
|
| D | nccl_all_reduce_thunk.cc | 90 int64 replica_count) { in GetNcclAllReduceConfig()
111 ThunkInfo thunk_info, mlir::lmhlo::AllReduceOp op, int64 replica_count, in NcclAllReduceThunk()
|
| /external/rust/crates/grpcio-sys/grpc/tools/run_tests/xds_k8s_test_driver/framework/ |
| D | xds_k8s_testcase.py | 198 def startTestServer(self, replica_count=1, **kwargs) -> XdsTestServer: argument
256 def startSecureTestServer(self, replica_count=1, **kwargs) -> XdsTestServer: argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 93 int replica_count = options.number_of_replicas(); in GetOrCreateLocalClient() local
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | op_stats.proto | 83 int32 replica_count = 9; field
|
| D | overview_page.proto | 184 int32 replica_count = 9; field
|
| /external/rust/crates/grpcio-sys/grpc/tools/run_tests/xds_k8s_test_driver/framework/test_app/ |
| D | server_app.py | 181 replica_count=1) -> XdsTestServer: argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 120 const string& name, int64 replica_count) { in CreateNewVerifiedModule()
129 int64 replica_count) { in ParseAndReturnVerifiedModule()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 378 int32 replica_count = 1; field
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 279 int64_t replica_count; member
|