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

/external/tensorflow/tensorflow/core/kernels/
H A Ddepthwise_conv_op.h201 const int64 input_scalar_size = args.in_depth % kPacketSize; local
244 for (int64 d = 0; d < input_scalar_size; ++d) {
266 in_buf += input_scalar_size * args.depth_multiplier;
H A Ddeep_conv2d.cc643 const int64 input_scalar_size = args.in_depth % kPacketSize; local
662 for (int64 d = 0; d < input_scalar_size; ++d) {

Completed in 496 milliseconds