Cross Reference: /external/tensorflow/tensorflow/core/ops/math_grad

Lines Matching defs:BatchMatMul

235   Tensor BatchMatMul(const Tensor& x, bool ax, const Tensor& y, bool ay) {
236     return MatMulCommon("BatchMatMul", "adj_x", "adj_y", x, ax, y, ay);
307     return MatMulGradCommon("BatchMatMul", "adj_x", "adj_y", x, ax, y, ay, dx,
945 // TODO{lukeiwanski}: Implement BatchMatMul for SYCL
955   test::ExpectClose(dx, BatchMatMul(dz, false, y, true));
956   test::ExpectClose(dy, BatchMatMul(x, true, dz, false));
967   test::ExpectClose(dx, BatchMatMul(dz, false, y, false));
968   test::ExpectClose(dy, BatchMatMul(dz, true, x, false));
979   test::ExpectClose(dx, BatchMatMul(y, false, dz, true));
980   test::ExpectClose(dy, BatchMatMul(x, false, dz, false));
991   test::ExpectClose(dx, BatchMatMul(y, true, dz, true));
992   test::ExpectClose(dy, BatchMatMul(dz, true, x, true));

OpenGrok