Searched refs:SparseMatMulOp (Results 1 – 3 of 3) sorted by relevance
100 class SparseMatMulOp : public MatMulOp { class102 explicit SparseMatMulOp(OpKernelConstruction* ctx) : MatMulOp(ctx, true) {} in SparseMatMulOp() function in tensorflow::__anondc94a7a40111::SparseMatMulOp104 ~SparseMatMulOp() override = default;107 REGISTER_XLA_OP(Name("SparseMatMul"), SparseMatMulOp);
954 class SparseMatMulOp : public OpKernel { class959 explicit SparseMatMulOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in SparseMatMulOp() function in tensorflow::SparseMatMulOp1081 TF_DISALLOW_COPY_AND_ASSIGN(SparseMatMulOp);1631 SparseMatMulOp<TA, TB, SparseMatMul>);1638 SparseMatMulOp<TA, TB, LibxsmmSparseMatMul>);
1197 SparseMatMulOp::getOperationName(), in LowerSparseMatMulOp()1203 auto op = cast<SparseMatMulOp>(src_op); in matchAndRewrite()