Searched defs:acc_buffer (Results 1 – 4 of 4) sorted by relevance
36 int32* acc_buffer) { in DepthwiseConvInitAccBuffer()71 int32 acc_buffer[kAccBufferMaxSize]; in DepthwiseConvHybridGeneral() local
491 auto acc_buffer = at::sparse_csr::create_acc_buffer<acc_t, scalar_t>( in reduce_sparse_csr_dim0_cuda_template() local582 auto acc_buffer = at::sparse_csr::create_acc_buffer<acc_t, scalar_t>( in reduce_sparse_csr_dim1_cuda_template() local
1077 auto acc_buffer = at::sparse_csr::create_acc_buffer<acc_t, scalar_t>( in reduce_sparse_csr_dim0_cpu_template() local1165 auto acc_buffer = at::sparse_csr::create_acc_buffer<acc_t, scalar_t>( in reduce_sparse_csr_dim1_cpu_template() local
1700 Vectorized<int32_t> acc_buffer[cb_size]; local