Searched refs:ExpandDimsGrad (Results 1 – 3 of 3) sorted by relevance
346 std::vector<Tensor> ExpandDimsGrad(const Tensor& x, const Tensor& s, in ExpandDimsGrad() function368 TEST(ArrayGradTest, ExpandDimsGrad) { in TEST() argument374 auto dx = ExpandDimsGrad(x, s, dy); in TEST()
213 Status ExpandDimsGrad(const Scope& scope, const Operation& op, in ExpandDimsGrad() function221 REGISTER_GRADIENT_OP("ExpandDims", ExpandDimsGrad);
184 TEST_F(ArrayGradTest, ExpandDimsGrad) { in TEST_F() argument