Searched defs:SparseTensorDenseMatMulFunctor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 83 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> { struct
85 Compute(OpKernelContext* ctx, typename TTypes<T>::Matrix out, in Compute()
|
| H A D | sparse_tensor_dense_matmul_op.cc | 327 struct SparseTensorDenseMatMulFunctor<CPUDevice, T, Tindices, ADJ_A, ADJ_B> { struct
328 static Status Compute(OpKernelContext* ctx, typename TTypes<T>::Matrix out, in Compute()
|
| H A D | sparse_tensor_dense_matmul_op.h | 31 struct SparseTensorDenseMatMulFunctor { struct
|