Searched defs:output_nnz (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op_gpu.cu.cc | 41 int64 output_nnz, int rank, int concat_dim, bool need_to_sort, in SparseConcatKernel()
85 int64 output_nnz, int rank, GpuDeviceArrayStruct<const T*> val_ptrs_data, in SparseConcatPermuteKernel()
158 const int64 output_nnz = nnz_sum; in operator ()() local
|
| D | sparse_slice_grad_op_gpu.cu.cc | 95 __global__ void SparseSliceGradKernel(int64_t input_nnz, int64_t output_nnz, in SparseSliceGradKernel()
141 const int64_t output_nnz = output_indices_mat.dimension(0); in operator ()() local
|
| D | sparse_slice_op_gpu.cu.cc | 89 int dims, int64_t output_nnz, in SparseSliceGatherKernel()
214 int64_t output_nnz = *output_nnz_host.data(); in operator ()() local
|
| D | sparse_concat_op.cc | 194 int64_t output_nnz = 0; in Compute() local
|