Searched refs:vectorized_size (Results 1 – 4 of 4) sorted by relevance
138 const int64 vectorized_size =140 const int64 scalar_size = filter_inner_dim_size - vectorized_size;143 const int64 padded_filter_stride = vectorized_size + kPacketSize;150 for (int64 j = 0; j < vectorized_size; j += kPacketSize) {156 padded_filter[output_base + vectorized_size + j] =157 filter[input_base + vectorized_size + j];161 padded_filter[output_base + vectorized_size + scalar_size + j] =
224 const int64 vectorized_size = (args.out_depth / kPacketSize) * kPacketSize; in CopyOutputBackpropRegion() local239 for (int64 d = 0; d < vectorized_size; d += kPacketSize) { in CopyOutputBackpropRegion()245 buffer[buf_base + vectorized_size + d] = out_bprop[vectorized_size + d]; in CopyOutputBackpropRegion()249 buffer[buf_base + vectorized_size + scalar_size + d] = in CopyOutputBackpropRegion()943 const int64 vectorized_size = (out_depth / kPacketSize) * kPacketSize; in operator ()() local944 const int64 scalar_size = out_depth - vectorized_size; in operator ()()953 for (int64 j = 0; j < vectorized_size; j += kPacketSize) { in operator ()()974 filter_backprop[output_base + vectorized_size + j] += in operator ()()975 output_buffer[buffer_base + vectorized_size + j]; in operator ()()
146 const int64 vectorized_size = args.in_depth / kPacketSize; in operator ()() local151 for (int64 d = 0; d < vectorized_size; ++d) { in operator ()()157 const int64 in_scalar_base = vectorized_size * input_stride; in operator ()()158 const int64 buf_scalar_base = vectorized_size * kPacketSize; in operator ()()223 const int64 vectorized_size = in_depth / kPacketSize; in operator ()() local242 for (int64 d = 0; d < vectorized_size; ++d) { in operator ()()248 const int64 scalar_base = vectorized_size * kPacketSize; in operator ()()
220 const Index vectorized_size = (last / PacketSize) * PacketSize;224 for (Index i = first * PacketSize; i < vectorized_size;228 for (Index i = vectorized_size + first; i < last; i += step_size) {