Searched refs:scalar_size (Results 1 – 5 of 5) sorted by relevance
140 const int64 scalar_size = filter_inner_dim_size - vectorized_size;142 const int64 pad_size = scalar_size > 0 ? kPacketSize - scalar_size : 0;155 for (int64 j = 0; j < scalar_size; ++j) {161 padded_filter[output_base + vectorized_size + scalar_size + j] =
213 const int64 scalar_size = args.out_depth % kPacketSize; in CopyOutputBackpropRegion() local214 const int64 pad_size = scalar_size > 0 ? kPacketSize - scalar_size : 0; in CopyOutputBackpropRegion()232 for (int64 d = 0; d < scalar_size; ++d) { in CopyOutputBackpropRegion()237 buffer[buf_base + vectorized_size + scalar_size + d] = in CopyOutputBackpropRegion()797 const int64 scalar_size = out_bprop_limit - out_bprop_index; in ComputeBackpropFilter() local798 for (int64 j = 0; j < scalar_size; ++j) { in ComputeBackpropFilter()903 const int64 scalar_size = out_depth - vectorized_size; in operator ()() local929 for (int64 j = 0; j < scalar_size; ++j) { in operator ()()
147 const int64 scalar_size = args.in_depth % kPacketSize; in operator ()() local159 for (int64 d = 0; d < scalar_size; ++d) { in operator ()()222 const int64 scalar_size = in_depth % kPacketSize; in operator ()() local249 for (int64 d = 0; d < scalar_size; ++d) { in operator ()()
374 inline size_t PaddingBytes(size_t buf_size, size_t scalar_size) { in PaddingBytes() argument375 return ((~buf_size) + 1) & (scalar_size - 1); in PaddingBytes()
637 fn padding_bytes(buf_size: usize, scalar_size: usize) -> usize { in padding_bytes()639 (!buf_size).wrapping_add(1) & (scalar_size.wrapping_sub(1)) in padding_bytes()