Searched refs:set_backend_config (Results 1 – 7 of 7) sorted by relevance
32 TF_RETURN_IF_ERROR(instr->set_backend_config(config)); in Run()
1309 Status set_backend_config(const tensorflow::protobuf::Message& proto);
2417 proto.set_backend_config(backend_config_); in ToProto()3163 Status HloInstruction::set_backend_config( in set_backend_config() function in xla::HloInstruction
126 TF_RETURN_IF_ERROR(custom_call->set_backend_config(options)); in CreateCholesky()
236 TF_RETURN_IF_ERROR(new_conv->set_backend_config(config)); in TryRewriteToCudnnForwardRelu()
394 TF_RETURN_IF_ERROR(new_call->set_backend_config(backend_config)); in RunOnInstruction()
539 custom_call->set_backend_config(GetDefaultBackendConfig())); in RunOnInstruction()