Searched refs:original_output (Results 1 – 4 of 4) sorted by relevance
384 for output, original_output in zip(flattened_output_tensors,390 original_output.shape))
3064 auto original_output = user_gte; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local3068 original_output = user_gte->users().back(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3069 unpadded_shape = original_output->shape(); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3071 for (auto u : original_output->users()) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3074 to_move.insert(original_output); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3216 outside_to_inside[original_output] = dus->mutable_operand(1); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()3217 add_users_if_available(original_output); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
269 original_output = root.f().numpy()271 self.assertNotEqual(original_output, imported_output)282 original_output = root.f().numpy()295 self.assertNotEqual(original_output, imported_output)
3537 int original_output = config.output;3558 config.output = original_output;