Home
last modified time | relevance | path

Searched refs:set_beta (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc66 gemm_config.set_beta(0.0); in HandleDot()
110 config.set_beta(1.0); in HandleAdd()
Dir_emitter_unnested.cc1267 config.backend_config.set_beta(gemm_bias.beta().convertToDouble()); in EmitGemmThunkFromMlir()
/external/tensorflow/tensorflow/core/kernels/
Dlrn_op.cc214 .set_beta(beta_); in launch()
267 .set_beta(beta_); in launch()
506 .set_beta(beta_); in launch()
581 .set_beta(beta_); in launch()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h890 NormalizeDescriptor& set_beta(float beta) { in set_beta() function