Searched defs:num_output_buffers (Results 1 – 6 of 6) sorted by relevance
134 let num_output_buffers: usize = in run() localVariable
266 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevicesInternal() local
442 int64_t num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionFitsInBudget() local
167 num_output_buffers: usize, field
350 pub fn num_output_buffers(&self) -> usize { in num_output_buffers() function
791 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local