Searched refs:SelectGrad (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad.cc | 666 Status SelectGrad(const AttrSlice& attrs, FunctionDef* g) { in SelectGrad() function
681 REGISTER_OP_GRADIENT("Select", SelectGrad);
|
| D | math_grad_test.cc | 362 void SelectGrad(const Tensor& c, const Tensor& x, const Tensor& y, Tensor* dc, in SelectGrad() function in tensorflow::__anona51524880111::MathGradTest
1121 SelectGrad(c, x, y, &dc, &dx, &dy); in TEST_F()
|