Home
last modified time | relevance | path

Searched refs:updated_config (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy.py629 updated_config = copy.deepcopy(config_proto)
631 updated_config.isolate_session_state = True
632 return updated_config
634 updated_config.isolate_session_state = False
640 del updated_config.device_filters[:]
642 updated_config.device_filters.extend(
645 updated_config.device_filters.append(
647 return updated_config
Dcollective_all_reduce_strategy.py689 updated_config = copy.deepcopy(config_proto)
693 rewrite_options = updated_config.graph_options.rewrite_options
705 updated_config.experimental.collective_nccl = True
708 return updated_config
715 updated_config.experimental.collective_group_leader = (
720 del updated_config.device_filters[:]
721 updated_config.device_filters.append(
724 return updated_config
Dmirrored_strategy.py692 updated_config = copy.deepcopy(config_proto)
693 updated_config.isolate_session_state = True
694 return updated_config
Dtpu_strategy.py1409 updated_config = copy.deepcopy(config_proto)
1410 updated_config.isolate_session_state = True
1413 updated_config.cluster_def.CopyFrom(cluster_spec.as_cluster_def())
1414 return updated_config
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc294 GemmBackendConfig updated_config = gemm_config; in RunOnInstruction() local
296 updated_config.set_selected_algorithm(*gemm_algorithm); in RunOnInstruction()
298 TF_RETURN_IF_ERROR(instr->set_backend_config(updated_config)); in RunOnInstruction()
299 return updated_config.SerializeAsString() != gemm_config.SerializeAsString(); in RunOnInstruction()