Searched refs:kPrimSliceGrad (Results 1 – 2 of 2) sorted by relevance
96 kernel_name_ == prim::kPrimSliceGrad->name() ? kSliceGradInputsNum : kStridedSliceGradInputsNum; in Launch()
209 inline const PrimitivePtr kPrimSliceGrad = std::make_shared<Primitive>("SliceGrad"); variable