Searched defs:FillDiagonal (Results 1 – 4 of 4) sorted by relevance
38 FillDiagonal() : BaseOperator(kNameFillDiagonal) { InitIOName({"input_x"}, {"y"}); } in FillDiagonal() function
22 __global__ void FillDiagonal(const size_t size, const float fill_value, const int64_t step_size, T … in FillDiagonal() function
5748 class FillDiagonal(Primitive): class
4789 [FillDiagonal] section4790 opInfo.subTypeOfInferShape=14791 opInfo.opsFlag=OPS_FLAG_CLOSE4792 opInfo.engine=DNN_VM_AICPU4793 opInfo.flagPartial=False4794 opInfo.computeCost=1004795 opInfo.flagAsync=False4796 opInfo.opKernelLib=CUSTAICPUKernel4797 opInfo.formatAgnostic=False4798 opInfo.userDefined=True[all …]