Searched refs:SqueezeGrad (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/ops/ | ||
H A D | array_grad.cc | 54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) { function in namespace:tensorflow 71 REGISTER_OP_GRADIENT("Squeeze", SqueezeGrad); |
H A D | array_grad_test.cc | 337 std::vector<Tensor> SqueezeGrad(const Tensor& x, const Tensor& dy) { function in namespace:tensorflow::__anon26715 356 TEST(ArrayGradTest, SqueezeGrad) { 361 auto dx = SqueezeGrad(x, dy); |
/external/tensorflow/tensorflow/cc/gradients/ | ||
H A D | array_grad.cc | 205 Status SqueezeGrad(const Scope& scope, const Operation& op, function in namespace:tensorflow::ops::__anon25351 212 REGISTER_GRADIENT_OP("Squeeze", SqueezeGrad); |
H A D | array_grad_test.cc | 184 TEST_F(ArrayGradTest, SqueezeGrad) { |
Completed in 614 milliseconds