Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_tensor_to_csr_sparse_matrix_op.cc151 functor::CalculateNNZPerBatchMatrixFromIndices<Device> in ComputeAsync()
292 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()(
295 extern template struct CalculateNNZPerBatchMatrixFromIndices<GPUDevice>;
Ddense_to_csr_sparse_matrix_op.cc195 functor::CalculateNNZPerBatchMatrixFromIndices<Device> in ComputeAsync()
372 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()(
375 extern template struct CalculateNNZPerBatchMatrixFromIndices<GPUDevice>;
Dkernels.h38 struct CalculateNNZPerBatchMatrixFromIndices { struct
Dkernels_gpu.cu.cc66 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()( in operator ()()