たれぱんのびぼーろく

わたしの備忘録、生物学とプログラミングが多いかも

2023-07-08から1日間の記事一覧

PyTorch AMP対応演算子

FP16へ autocast されるCUDA Ops: Conv系 Conv系 conv1d conv2d conv3d ConvT系 conv_transpose1d conv_transpose2d conv_transpose3d RNN系 RNNCell LSTMCell GRUCell Linear系 linear matmul __matmul__ chain_matmul baddbmm addbmm addmm addmv addr bmm…