Searched refs:kAttrHiddenSize (Results 1 – 3 of 3) sorted by relevance
477 constexpr auto kAttrHiddenSize = "hidden_size"; variable
232 AnfAlgo::CopyNodeAttr(kAttrHiddenSize, node, trans_data); in AddTransOpNodeToGraph()
865 …if (!AnfAlgo::HasNodeAttr(kAttrHiddenSize, cnode) || !AnfAlgo::HasNodeAttr(kAttrInputSize, cnode))… in GetAttrInputAndHiddenSize()871 input_hidden_size[1] = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrHiddenSize); in GetAttrInputAndHiddenSize()