Searched refs:pack_a (Results 1 – 2 of 2) sorted by relevance
143 fbgemm::PackAWithQuantRowOffset<uint8_t> pack_a( in fbgemm_linear_int8_weight_fp32_activation() local167 /*row_offsets=*/pack_a.getRowOffsetBuffer(), in fbgemm_linear_int8_weight_fp32_activation()173 /*packA=*/pack_a, in fbgemm_linear_int8_weight_fp32_activation()
565 ITensorPack pack_a = in run() local570 …().schedule_op(_mtx_a_reshape_kernel.get(), Window::DimY, _mtx_a_reshape_kernel->window(), pack_a); in run()