Home
last modified time | relevance | path

Searched defs:indices_row (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_gpu.cu.cc42 Tindex indices_row = grad_index / col_size; in SparseApplyAdagradKernel() local
81 Tindex indices_row = grad_index / col_size; in SparseApplyProximalAdagradKernel() local
124 const Tindex indices_row = grad_index / col_size; in SparseApplyFtrlKernel() local
228 Tindex indices_row = grad_index / col_size; in SparseApplyKerasMomentumKernel() local
808 Tindex indices_row = grad_index / col_size; in SparseApplyAdadeltaKernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSoftMax.cpp94 auto indices_row = indices_accessor[j]; in get_offsets() local
143 const auto indices_row = indices_accessor[j]; in get_pools() local