| 889 | REGISTER_OP_GRADIENT("BatchMatMul", BatchMatMulGrad); |
| 890 | |
| 891 | Status BatchMatMulV2Grad(const AttrSlice& attrs, FunctionDef* g) { |
| 892 | return MatMulGradCommon("BatchMatMulV2", "adj_x", "adj_y", attrs, g, |
| 893 | true /* enable_broadcasting */); |
| 894 | } |
| 895 | REGISTER_OP_GRADIENT("BatchMatMulV2", BatchMatMulV2Grad); |
| 896 | |
| 897 | // REGISTER_OP_GRADIENT("SparseMatMul", SparseMatMulGrad); |