Home
last modified time | relevance | path

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

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