Searched refs:BatchMatMulV2Grad (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad_test.cc | 356 void BatchMatMulV2Grad(const Tensor& x, bool ax, const Tensor& y, bool ay, in BatchMatMulV2Grad() function in tensorflow::__anona51524880111::MathGradTest
1236 BatchMatMulV2Grad(x, false, y, false, &dx, &dy); in TEST_F()
1248 BatchMatMulV2Grad(x, false, y, true, &dx, &dy); in TEST_F()
1260 BatchMatMulV2Grad(x, true, y, false, &dx, &dy); in TEST_F()
1272 BatchMatMulV2Grad(x, true, y, true, &dx, &dy); in TEST_F()
1286 BatchMatMulV2Grad(x, false, y, false, &dx, &dy); in TEST_F()
1307 BatchMatMulV2Grad(x, false, y, false, &dx, &dy); in TEST_F()
1325 BatchMatMulV2Grad(x, false, y, false, &dx, &dy); in TEST_F()
1347 BatchMatMulV2Grad(x, true, y, true, &dx, &dy); in TEST_F()
|
| D | math_grad.cc | 910 Status BatchMatMulV2Grad(const AttrSlice& attrs, FunctionDef* g) { in BatchMatMulV2Grad() function
914 REGISTER_OP_GRADIENT("BatchMatMulV2", BatchMatMulV2Grad);
|