Home
last modified time | relevance | path

Searched defs:new_accum (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc567 auto new_accum = accum + grad.square(); in operator ()() local
602 auto new_accum = accum + grad.square(); in operator ()() local
639 auto new_accum = accum + grad.square(); in operator ()() local
679 auto new_accum = accum + grad.square(); in operator ()() local
H A Dtraining_ops.cc357 auto new_accum = accum + grad * grad; in operator ()() local
398 auto new_accum = accum + grad * grad; in operator ()() local
437 auto new_accum = accum + grad.square(); in operator ()() local
475 auto new_accum = accum + grad.square(); in operator ()() local
538 auto new_accum = accum + grad.square(); in ComputeFtrl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc840 xla::XlaOp new_accum = accum + xla::Square(grad); in CompileFtrl() local