Home
last modified time | relevance | path

Searched defs:lr_power (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dtraining_ops_test.py93 lr_power=-0.5): argument
138 lr_power=-0.5): argument
253 lr_power=-0.5): argument
292 lr_power=-0.5): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/
H A Doptimization_parameters.proto158 float lr_power = 3; field
243 float lr_power = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dftrl_ops_test.py36 lr_power=1, multiply_linear_by_lr=False): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops.cc507 const T& l2, const T& lr_power, in FtrlCompute()
2656 const Tensor& lr_power = ctx->input(lr_power_index); in Compute() local
2882 const Tensor& lr_power = ctx->input(lr_power_index); in Compute() local
H A Dtraining_ops_gpu.cu.cc117 const T* l2_shrinkage, const T* lr_power, in SparseApplyFtrlKernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc815 xla::XlaOp lr_power; in CompileFtrl() local