Searched refs:output_vectorized_size (Results 1 – 2 of 2) sorted by relevance
92 const int64 output_vectorized_size = in Run() local96 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in Run()122 output_vectorized_size + j * padded_filter_inner_dim_size; in Run()134 base_output_index + output_vectorized_size; in Run()
290 const int64 output_vectorized_size = (out_depth / kPacketSize) * kPacketSize; in ComputeBackpropInput() local302 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropInput()328 output_vectorized_size + j * padded_filter_inner_dim_size; in ComputeBackpropInput()339 output[base_output_index + output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()344 out_buffer[output_vectorized_size + j] = out_buf[j]; in ComputeBackpropInput()756 const int64 output_vectorized_size = in ComputeBackpropFilter() local769 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropFilter()790 for (int i = 0; i < output_vectorized_size; i += kPacketSize) { in ComputeBackpropFilter()