Searched refs:attr_adj_x (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad.cc | 811 const string& attr_adj_x, 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()
868 Status MatMulGradCommon(const string& opname, const string& attr_adj_x, in MatMulGradCommon() argument
879 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, attr_adj_x, &ta)); 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 | 251 Tensor MatMulCommon(const string& opname, const string& attr_adj_x, in MatMulCommon() argument
260 {{"T", T}, {attr_adj_x, ax}, {attr_adj_y, ay}}), in MatMulCommon()
284 void MatMulGradCommon(const string& opname, const string& attr_adj_x, 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
1122 TF_RETURN_IF_ERROR(GetNodeAttr(product.node()->attrs(), attr_adj_x, &ta)); in MatMulGradCommon()
|