Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h200 const int64 input_vectorized_size =
202 const int64 input_scalar_size = args.in_depth - input_vectorized_size;
315 for (int64 d = 0; d < input_vectorized_size; d += kPacketSize) {
321 T v = in[input_vectorized_size + d];
Ddeep_conv2d.cc642 const int64 input_vectorized_size = in operator ()() local
657 for (int64 d = 0; d < input_vectorized_size; d += kPacketSize) { in operator ()()
664 tile[d] = in[input_vectorized_size + d]; in operator ()()