Searched defs:flatten_inputs (Results 1 – 2 of 2) sorted by relevance
331 …auto flatten_inputs = PyNativeAlgo::DataConvert::FlattenTensorSeqInValueSeq(grad_param->op_grad_in… in KPynativeOp() local421 …auto flatten_inputs = PyNativeAlgo::DataConvert::FlattenTensorSeqInValueSeq(grad_param->op_grad_in… in BuildGraphBackwardNode() local531 … const ValuePtrList &flatten_inputs, const OpGradInfoPtr &op_grad_info) { in BuildFuncBackwardNode()540 …Ptr FuncGrad::BuildCustomBackwardNode(const PrimitivePtr &prim, const ValuePtrList &flatten_inputs, in BuildCustomBackwardNode()565 …dePtr FuncGrad::BuildHookBackwardNode(const PrimitivePtr &prim, const ValuePtrList &flatten_inputs, in BuildHookBackwardNode()577 …dePtr FuncGrad::BuildFakeBackwardNode(const PrimitivePtr &prim, const ValuePtrList &flatten_inputs, in BuildFakeBackwardNode()
52 std::vector<AnfNodePtr> flatten_inputs = {NewValueNode(prim::kPrimFlatten), input_node}; in NewFlattenNode() local