Searched refs:flatten_grad (Results 1 – 5 of 5) sorted by relevance
35 auto flatten_grad = std::make_shared<FlattenGrad>(); in TEST_F() local39 auto abstract = flatten_grad->Infer({tensor_x->ToAbstract(), in_shape->ToAbstract()}); in TEST_F()
37 setattr(vm, 'flatten_grad', flatten_grad)
267 dout = vm.flatten_grad(dout, x)
343 def flatten_grad(dout, x): function
215 flatten_grad = P.Reshape()218 dx = flatten_grad(dout, shape_op(x))