a ==icã@sHddd„Zdd„Zdd„Zdd„Zd d „Zd d „Zdd d„Zdd„ZdS)FcKs"|r d|d<|jj||fi|¤ŽS)z Transpose.ZCUDNNÚengine)ÚnetZ Transpose)ÚmodelÚblob_inÚblob_outZ use_cudnnÚkwargs©rún/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/helpers/algebra.pyÚ transpose sr cKs|jj||fi|¤ŽS)ÚSum)rr ©rrrrrrrÚsumsr cKs|jj||fi|¤ŽS)Ú ReduceSum)rr r rrrÚ reduce_sumsrcKs|jj||fi|¤ŽS)ZSubtract)rÚSubr rrrÚsubsrcKs|jj||fi|¤ŽS)zMatrix multiplication)rZMatMulr rrrÚmat_mulsrcKs|jj||fi|¤ŽS)ÚArgMin)rrr rrrÚarg_min$srcKs"|r d|d<|jj||fi|¤ŽS)NZ TENSORCOREr)rZ BatchMatMul)rrrZenable_tensor_corerrrrÚ batch_mat_mul(srcKs|jj||fi|¤ŽS)N)rZ!SparseLengthsSum4BitRowwiseSparser rrrÚ&sparse_lengths_sum_4bit_rowwise_sparse/srN)F)F)r r rrrrrrrrrrÚ s ÿ