Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_op.h202 const int64 input_scalar_size = args.in_depth - input_vectorized_size;
320 for (int64 d = 0; d < input_scalar_size; ++d) {
324 in_buf += input_scalar_size;
Ddeep_conv2d.cc644 const int64 input_scalar_size = args.in_depth % kPacketSize; in operator ()() local
663 for (int64 d = 0; d < input_scalar_size; ++d) { in operator ()()