Searched refs:SparseMatMulOp (Results 1 – 2 of 2) sorted by relevance
99 class SparseMatMulOp : public MatMulOp { class101 explicit SparseMatMulOp(OpKernelConstruction* ctx) : MatMulOp(ctx, true) {} in SparseMatMulOp() function in tensorflow::__anon52c92e4d0111::SparseMatMulOp103 ~SparseMatMulOp() override = default;106 REGISTER_XLA_OP(Name("SparseMatMul"), SparseMatMulOp);
955 class SparseMatMulOp : public OpKernel { class960 explicit SparseMatMulOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in SparseMatMulOp() function in tensorflow::SparseMatMulOp1082 TF_DISALLOW_COPY_AND_ASSIGN(SparseMatMulOp);1629 SparseMatMulOp<TA, TB, SparseMatMul>);1636 SparseMatMulOp<TA, TB, LibxsmmSparseMatMul>);