Home
last modified time | relevance | path

Searched refs:dm_vectorized_size (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h206 const int64 dm_vectorized_size =
247 if (dm_vectorized_size > 0) {
252 for (int64 dm = 0; dm < dm_vectorized_size; dm += kPacketSize) {
257 in_buf[base + dm_vectorized_size + dm] = v;
Ddepthwise_conv_grad_op.cc298 const int64 dm_vectorized_size = in ComputeBackpropInput() local
354 for (int64 dm = 0; dm < dm_vectorized_size; dm += kPacketSize) { in ComputeBackpropInput()
360 accum += out_buffer[index + dm_vectorized_size + dm]; in ComputeBackpropInput()