Searched refs:output_scalar_size (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Ddepthwise_conv_op.h211 const int64 output_scalar_size = args.out_depth % kPacketSize; local
213 output_scalar_size > 0 ? kPacketSize - output_scalar_size : 0;
H A Ddepthwise_conv_op.cc90 const int64 output_scalar_size = out_depth % kPacketSize; local
117 if (output_scalar_size > 0) {
134 for (int j = 0; j < output_scalar_size; ++j) {
H A Ddepthwise_conv_grad_op.cc288 const int64 output_scalar_size = out_depth % kPacketSize; local
321 if (output_scalar_size > 0) {
335 for (int j = 0; j < output_scalar_size; ++j) {
340 for (int j = 0; j < output_scalar_size; ++j) {

Completed in 185 milliseconds