Searched refs:kGradIndex (Results 1 – 4 of 4) sorted by relevance
35 auto grad_cast = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), kGradIndex); in Process()54 outputs_type[kGradIndex] = AnfAlgo::GetPrevNodeOutputInferDataType(grad_cast, 0); in Process()
79 constexpr int64_t kGradIndex = 0; variable
728 …if (attrs.count(kGradIndex) == 0 || attrs.count(kIndiceIndex) == 0 || attrs.count(kFirstDimSize) =… in SparsePartitioner()732 auto iter = attrs.find(kGradIndex); in SparsePartitioner()
49 constexpr size_t kGradIndex = 3; variable