Searched refs:output_scalar_size (Results 1 – 3 of 3) sorted by relevance
205 const int64 output_scalar_size = args.out_depth % kPacketSize;207 output_scalar_size > 0 ? kPacketSize - output_scalar_size : 0;
96 const int64 output_scalar_size = out_depth % kPacketSize; in Run() local123 if (output_scalar_size > 0) { in Run()140 for (int j = 0; j < output_scalar_size; ++j) { in Run()
303 const int64 output_scalar_size = out_depth % kPacketSize; in ComputeBackpropInput() local336 if (output_scalar_size > 0) { in ComputeBackpropInput()350 for (int j = 0; j < output_scalar_size; ++j) { in ComputeBackpropInput()355 for (int j = 0; j < output_scalar_size; ++j) { in ComputeBackpropInput()