Searched refs:kAttrInputSize (Results 1 – 4 of 4) sorted by relevance
41 constexpr auto kAttrInputSize = "input_size"; variable167 AnfAlgo::CopyNodeAttr(kAttrInputSizes, kAttrInputSize, conv2d_backin, depth_conv_backin); in CreateDepthwiseConv2DBackpropInput()
478 constexpr auto kAttrInputSize = "input_size"; variable
233 AnfAlgo::CopyNodeAttr(kAttrInputSize, node, trans_data); in AddTransOpNodeToGraph()
865 …if (!AnfAlgo::HasNodeAttr(kAttrHiddenSize, cnode) || !AnfAlgo::HasNodeAttr(kAttrInputSize, cnode))… in GetAttrInputAndHiddenSize()870 input_hidden_size[0] = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrInputSize); in GetAttrInputAndHiddenSize()