Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc190 instr.set_activation(activation_mode); in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc792 auto set_activation = [&, this](auto op) -> Status { in EmitDnnConvolution() local
828 TF_RETURN_IF_ERROR(set_activation(cnn_fused)); in EmitDnnConvolution()
839 TF_RETURN_IF_ERROR(set_activation(cnn_fused_side_input)); in EmitDnnConvolution()