Searched refs:output_scalar_size (Results 1 – 3 of 3) sorted by relevance
211 const int64 output_scalar_size = args.out_depth % kPacketSize;213 output_scalar_size > 0 ? kPacketSize - output_scalar_size : 0;
91 const int64 output_scalar_size = out_depth % kPacketSize; in Run() local118 if (output_scalar_size > 0) { in Run()135 for (int j = 0; j < output_scalar_size; ++j) { in Run()
291 const int64 output_scalar_size = out_depth % kPacketSize; in ComputeBackpropInput() local324 if (output_scalar_size > 0) { in ComputeBackpropInput()338 for (int j = 0; j < output_scalar_size; ++j) { in ComputeBackpropInput()343 for (int j = 0; j < output_scalar_size; ++j) { in ComputeBackpropInput()