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.cc152 functor::CalculateNNZPerBatchMatrixFromIndices<Device> in ComputeAsync()
293 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()(
296 extern template struct CalculateNNZPerBatchMatrixFromIndices<GPUDevice>;
Ddense_to_csr_sparse_matrix_op.cc196 functor::CalculateNNZPerBatchMatrixFromIndices<Device> in ComputeAsync()
371 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()(
374 extern template struct CalculateNNZPerBatchMatrixFromIndices<GPUDevice>;
Dkernels.h38 struct CalculateNNZPerBatchMatrixFromIndices { struct
Dkernels_gpu.cu.cc53 Status CalculateNNZPerBatchMatrixFromIndices<GPUDevice>::operator()( in operator ()()