Searched defs:MatrixBandPart (Results 1 – 4 of 4) sorted by relevance
42 MatrixBandPart() : BaseOperator(kNameMatrixBandPart) { InitIOName({"x"}, {"y"}); } in MatrixBandPart() function
88 cudaError_t MatrixBandPart(const size_t output_outer_size, const T *x_ptr, const size_t m, const si… in MatrixBandPart() function
1121 class MatrixBandPart(Primitive): class
6088 [MatrixBandPart] section6089 opInfo.subTypeOfInferShape=16090 opInfo.opsFlag=OPS_FLAG_CLOSE6091 opInfo.engine=DNN_VM_AICPU6092 opInfo.flagPartial=False6093 opInfo.computeCost=1006094 opInfo.flagAsync=False6095 opInfo.opKernelLib=CUSTAICPUKernel6096 opInfo.formatAgnostic=False6097 opInfo.userDefined=True[all …]