Searched refs:c_config (Results 1 – 6 of 6) sorted by relevance
453 xla::HloModuleConfig FromC(const XLA_HloModuleConfig& c_config) { in FromC() argument454 xla::HloModuleConfig config = c_config.has_entry_computation_layout in FromC()455 ? ConfigWithLayout(c_config) in FromC()457 config.set_launch_id(c_config.launch_id); in FromC()458 config.set_seed(c_config.seed); in FromC()459 config.set_replica_count(c_config.replica_count); in FromC()460 config.set_num_partitions(c_config.num_partitions); in FromC()461 config.set_use_spmd_partitioning(c_config.use_spmd_partitioning); in FromC()462 if (c_config.has_static_device_assignment) { in FromC()465 c_config.static_device_assignment)); in FromC()[all …]
90 SE_StreamExecutorConfig* c_config = in GetUncachedExecutor() local93 tpu::ExecutorApiFn()->TpuStreamExecutorConfig_SetOrdinalFn(c_config, in GetUncachedExecutor()98 platform_, c_config, status.c_status); in GetUncachedExecutor()99 tpu::ExecutorApiFn()->TpuStreamExecutorConfig_FreeFn(c_config); in GetUncachedExecutor()
112 xla::HloModuleConfig FromC(const XLA_HloModuleConfig& c_config);113 void Free(XLA_HloModuleConfig* c_config);
139 grpc_tls_server_authorization_check_config* c_config() const { in c_config() function
172 options, config->c_config()); in set_server_authorization_check_config()