Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_concat_op_gpu.cu.cc117 struct SparseConcatFunctor<GPUDevice, T> { struct
118 void operator()(OpKernelContext* context, const OpInputList& inds, in operator ()()
241 Status allocate_outputs(OpKernelContext* context, int rank, int64 output_nnz, in allocate_outputs()
H A Dsparse_concat_op.cc42 struct SparseConcatFunctor<CPUDevice, T> { struct
43 void operator()(OpKernelContext* context, const OpInputList& inds, in operator ()()
H A Dsparse_concat_op.h26 struct SparseConcatFunctor { struct