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.h201 const int64 input_scalar_size = args.in_depth % kPacketSize;
244 for (int64 d = 0; d < input_scalar_size; ++d) {
266 in_buf += input_scalar_size * args.depth_multiplier;
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 ()()