Home
last modified time | relevance | path

Searched refs:COOSparseMatrixToSparseTensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/sparse/
Dcsr_sparse_matrix_to_sparse_tensor_op.cc192 functor::COOSparseMatrixToSparseTensor<Device> coo_to_st; in Compute()
225 struct COOSparseMatrixToSparseTensor<GPUDevice> { struct
233 extern template struct COOSparseMatrixToSparseTensor<GPUDevice>;
Dcsr_sparse_matrix_to_dense_op.cc186 functor::COOSparseMatrixToSparseTensor<Device> coo_to_st; in Compute()
239 struct COOSparseMatrixToSparseTensor<GPUDevice> { struct
247 extern template struct COOSparseMatrixToSparseTensor<GPUDevice>;
Dkernels.h84 struct COOSparseMatrixToSparseTensor { struct
Dkernels_gpu.cu.cc217 Status COOSparseMatrixToSparseTensor<GPUDevice>::operator()( in operator ()()