Searched defs:output_vec (Results 1 – 11 of 11) sorted by relevance
312 Vec output_vec = Vec::loadu(output_ptr + d1); in _vec_softmax_backward() local333 Vec output_vec = Vec::loadu(output_ptr + d2); in _vec_softmax_backward() local411 Vec output_vec = Vec::loadu(output_ptr + d1); in _vec_softmax_backward() local542 Vec output_vec = Vec::loadu(output_ptr + d2); in _vec_log_softmax_backward() local645 Vec output_vec = Vec::loadu(output_ptr + d2); in _vec_log_softmax_backward() local831 Vec output_vec = in _vec_softmax() local838 Vec output_vec = in _vec_softmax() local
111 Vec output_vec = data_vec * alpha_vec + beta_vec; in batch_norm_cpu_contiguous_impl() local116 Vec output_vec = data_vec * alpha_vec + beta_vec; in batch_norm_cpu_contiguous_impl() local161 Vec output_vec = data_vec * alpha_vec + beta_vec; in batch_norm_cpu_channels_last_impl() local168 Vec output_vec = data_vec * alpha_vec + beta_vec; in batch_norm_cpu_channels_last_impl() local
158 Vec output_vec=radius * theta.cos() * std + mean; in normal_fill_16_VSX() local
76 Vec output_vec = min_vec - buffer_vec.log1p(); in log_sigmoid_cpu_kernel() local84 Vec output_vec = min_vec - buffer_vec.log1p(); in log_sigmoid_cpu_kernel() local
228 let mut output_vec = in encrypt() localVariable296 let mut output_vec = in decrypt() localVariable
54 std::vector<uint32_t> output_vec; in PackToUint32() local
990 let output_vec = Arc::new(Mutex::new(vec![])); in run_queue_test() localVariable1035 let mut output_vec = output_vec.lock(); in run_queue_test() localVariable
299 auto output_vec = _mm256_cvtepi32_ps(input_vec); in convert() local319 auto output_vec = _mm256_cvtepi32_pd(input_128_vec); in convert() local
289 auto output_vec = _mm256_cvtepi32_ps(input_vec); in convert() local309 auto output_vec = _mm256_cvtepi32_pd(input_128_vec); in convert() local
321 auto output_vec = _mm512_cvtepi32_ps(input_vec); in convert() local341 auto output_vec = _mm512_cvtepi32_pd(input_256_vec); in convert() local