Searched defs:original_input (Results 1 – 3 of 3) sorted by relevance
70 const MaybeOwningDeviceMemory& original_input = in AllocateOutputMemoryWithInputReuse() local
531 Value original_input = strided_slice_op.input(); in RewriteNewAxisMask() local
1305 HloInstruction* original_input = hlo->mutable_operand(operand_index); in HandleConditional() local