Searched defs:FlattenGrad (Results 1 – 2 of 2) sorted by relevance
31 FlattenGrad() : BaseOperator(kNameFlattenGrad) { InitIOName({"x", "shape"}, {"output"}); } in FlattenGrad() function
477 class FlattenGrad(PrimitiveWithInfer): class