Searched refs:attr_adj_y (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad.cc | 812 const string& attr_adj_y, const string& x0, in MatMulGradHelper() argument
822 {{"T", "$T"}, {attr_adj_x, ax0}, {attr_adj_y, ax1}}}, in MatMulGradHelper()
826 {{"T", "$T"}, {attr_adj_x, ay0}, {attr_adj_y, ay1}}}, in MatMulGradHelper()
869 const string& attr_adj_y, const AttrSlice& attrs, in MatMulGradCommon() argument
880 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, attr_adj_y, &tb)); in MatMulGradCommon()
882 return MatMulGradHelper(g, opname, attr_adj_x, attr_adj_y, "dz", false, "y", in MatMulGradCommon()
886 return MatMulGradHelper(g, opname, attr_adj_x, attr_adj_y, "dz", false, "y", in MatMulGradCommon()
890 return MatMulGradHelper(g, opname, attr_adj_x, attr_adj_y, "y", false, "dz", in MatMulGradCommon()
894 return MatMulGradHelper(g, opname, attr_adj_x, attr_adj_y, "y", true, "dz", in MatMulGradCommon()
|
| D | math_grad_test.cc | 252 const string& attr_adj_y, const Tensor& x, bool ax, in MatMulCommon() argument
260 {{"T", T}, {attr_adj_x, ax}, {attr_adj_y, ay}}), in MatMulCommon()
285 const string& attr_adj_y, const Tensor& x, bool ax, in MatMulGradCommon() argument
298 {{"T", T}, {attr_adj_x, ax}, {attr_adj_y, ay}}}, in MatMulGradCommon()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 1109 const string& attr_adj_x, const string& attr_adj_y, in MatMulGradCommon() argument
1123 TF_RETURN_IF_ERROR(GetNodeAttr(product.node()->attrs(), attr_adj_y, &tb)); in MatMulGradCommon()
|