Searched refs:TransposeGrad (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_grad.cc | 338 Status TransposeGrad(const AttrSlice& attrs, FunctionDef* g) { in TransposeGrad() function
355 REGISTER_OP_GRADIENT("Transpose", TransposeGrad);
|
| D | array_grad_test.cc | 414 std::vector<Tensor> TransposeGrad(const Tensor& x, const Tensor& p, in TransposeGrad() function
436 TEST(ArrayGradTest, TransposeGrad) { in TEST() argument
442 auto dx = TransposeGrad(x, p, dy); in TEST()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 232 Status TransposeGrad(const Scope& scope, const Operation& op, in TransposeGrad() function
240 REGISTER_GRADIENT_OP("Transpose", TransposeGrad);
|
| D | array_grad_test.cc | 200 TEST_F(ArrayGradTest, TransposeGrad) { in TEST_F() argument
|