Home
last modified time | relevance | path

Searched defs:SparseMatMulOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmatmul_op.cc104 class SparseMatMulOp : public MatMulOp { class
106 explicit SparseMatMulOp(OpKernelConstruction* ctx) : MatMulOp(ctx, true) {} in SparseMatMulOp() function in tensorflow::__anonbe5def1a0111::SparseMatMulOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_matmul_op.cc950 class SparseMatMulOp : public OpKernel { class
955 explicit SparseMatMulOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in SparseMatMulOp() function in tensorflow::SparseMatMulOp