Searched refs:output_slices (Results 1 – 3 of 3) sorted by relevance
63 const int64 output_slices = channel_count; in Compute() local69 0, TensorShape({output_slices, output_height, output_width}), in Compute()
249 output_slices, old_output_slices): argument280 graph.structured_outputs[structured_idx] = output_slices
400 std::vector<HloInstruction*> output_slices(concated_outputs.size() * in CreateFusedComputation() local411 output_slices[concated_outputs.size() * j + i], in CreateFusedComputation()419 auto tuple = comp->AddInstruction(HloInstruction::CreateTuple(output_slices)); in CreateFusedComputation()