Home
last modified time | relevance | path

Searched defs:sparse_handle (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensors_map_ops.cc201 Tensor sparse_handle(DT_INT64, TensorShape({})); in Compute() local
/external/tensorflow/tensorflow/core/util/
Drocm_sparse.cc457 SparseFnT op, OpKernelContext* context, hipsparseHandle_t sparse_handle, in CsrgeamBufferSizeExtImpl()
Dcuda_sparse.cc721 SparseFnT op, OpKernelContext* context, cusparseHandle_t sparse_handle, in CsrgeamBufferSizeExtImpl()