Home
last modified time | relevance | path

Searched refs:set_activation_mode (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc247 config.set_activation_mode( in TryRewriteToCudnnForwardRelu()
Dir_emitter_unnested.cc1186 auto set_activation_mode = [&](auto op) -> Status { in EmitConvolutionThunkFromMlir() local
1189 descriptor.backend_config.set_activation_mode( in EmitConvolutionThunkFromMlir()
1206 TF_RETURN_IF_ERROR(set_activation_mode(op)); in EmitConvolutionThunkFromMlir()
1210 TF_RETURN_IF_ERROR(set_activation_mode(op)); in EmitConvolutionThunkFromMlir()