Searched defs:Diagonal (Results 1 – 7 of 7) sorted by relevance
188 real64 Diagonal () const in Diagonal() function331 real64 Diagonal () const in Diagonal() function
279 class Diagonal(nn.Cell): class
20 __global__ void Diagonal(const size_t out_size, const T *input, const size_t *inp_shape, const size… in Diagonal() function
264 void Diagonal(T *input_ptr, T *output_ptr, const std::vector<size_t> &inp_shape, in Diagonal() function
725 Diagonal() : BaseOperator(kNameDiagonal) {} in Diagonal() function
5543 [Diagonal] section5544 opInfo.subTypeOfInferShape=15545 opInfo.opsFlag=OPS_FLAG_CLOSE5546 opInfo.engine=DNN_VM_AICPU5547 opInfo.flagPartial=False5548 opInfo.computeCost=1005549 opInfo.flagAsync=False5550 opInfo.opKernelLib=CUSTAICPUKernel5551 opInfo.formatAgnostic=False5552 opInfo.kernelSo=libcust_cpu_kernels.so[all …]