Searched refs:num_output_buffers (Results 1 – 3 of 3) sorted by relevance
171 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local173 outputs.resize(num_output_buffers); in ExecuteShardedOnLocalDevices()174 for (int buffer_id = 0; buffer_id < num_output_buffers; ++buffer_id) { in ExecuteShardedOnLocalDevices()
396 int64 num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionWouldBeTooLarge() local407 num_output_buffers <= in FusionWouldBeTooLarge()415 << " and num_output_buffers = " << num_output_buffers in FusionWouldBeTooLarge()439 return operands.size() + num_output_buffers > kMaxOperandsAndOutputsPerFusion; in FusionWouldBeTooLarge()
781 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local783 outputs.resize(num_output_buffers); in ExecuteShardedOnLocalDevices()784 for (int buffer_id = 0; buffer_id < num_output_buffers; ++buffer_id) { in ExecuteShardedOnLocalDevices()