Searched refs:SqueezeGrad (Results 1 – 4 of 4) sorted by relevance
54 Status SqueezeGrad(const AttrSlice& attrs, FunctionDef* g) { in SqueezeGrad() function71 REGISTER_OP_GRADIENT("Squeeze", SqueezeGrad);
385 std::vector<Tensor> SqueezeGrad(const Tensor& x, const Tensor& dy) { in SqueezeGrad() function404 TEST(ArrayGradTest, SqueezeGrad) { in TEST() argument409 auto dx = SqueezeGrad(x, dy); in TEST()
223 Status SqueezeGrad(const Scope& scope, const Operation& op, in SqueezeGrad() function230 REGISTER_GRADIENT_OP("Squeeze", SqueezeGrad);
192 TEST_F(ArrayGradTest, SqueezeGrad) { in TEST_F() argument