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.h201 const int64_t input_vectorized_size =
203 const int64_t input_scalar_size = args.in_depth - input_vectorized_size;
316 for (int64_t d = 0; d < input_vectorized_size; d += kPacketSize) {
322 T v = in[input_vectorized_size + d];
Ddeep_conv2d.cc653 const int64_t input_vectorized_size = in operator ()() local
668 for (int64_t d = 0; d < input_vectorized_size; d += kPacketSize) { in operator ()()
675 tile[d] = in[input_vectorized_size + d]; in operator ()()