Searched defs:output_reshape (Results 1 – 2 of 2) sorted by relevance
90 HloInstruction* output_reshape = instruction->parent()->AddInstruction( in ExpandInstruction() local
101 const HloInstruction* output_reshape = entry_computation->root_instruction(); in TEST_F() local