Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_reorder_op.cc44 struct SparseReorderFunctor<CPUDevice, T> { struct
45 void operator()(OpKernelContext* context, const Tensor& input_ind, in operator ()()
H A Dsparse_reorder_op_gpu.cu.cc69 struct SparseReorderFunctor<GPUDevice, T> { struct
70 void operator()(OpKernelContext* c, const Tensor& input_ind, in operator ()()
H A Dsparse_reorder_op.h26 struct SparseReorderFunctor { struct