Searched refs:output_vectorized_size (Results 1 – 2 of 2) sorted by relevance
97 const int64 output_vectorized_size = in Run() local101 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in Run()127 output_vectorized_size + j * padded_filter_inner_dim_size; in Run()139 base_output_index + output_vectorized_size; in Run()
302 const int64 output_vectorized_size = (out_depth / kPacketSize) * kPacketSize; in ComputeBackpropInput() local314 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropInput()340 output_vectorized_size + j * padded_filter_inner_dim_size; in ComputeBackpropInput()351 output[base_output_index + output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()356 out_buffer[output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()797 const int64 output_vectorized_size = in ComputeBackpropFilter() local810 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropFilter()831 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropFilter()