Searched defs:oSizes (Results 1 – 2 of 2) sorted by relevance
105 auto oSizes = qx.sizes(); in q_batch_norm1d_impl() local206 auto oSizes = qx.sizes(); in q_batch_norm2d_impl() local302 auto oSizes = qx.sizes(); in q_batch_norm3d_impl() local
247 std::vector<int64_t> oSizes; in q_maxpool_2d() local411 std::vector<int64_t> oSizes = {nbatch, oC, oT, oH, oW}; in q_maxpool_3d() local