Searched refs:shifted_normalized (Results 1 – 1 of 1) sorted by relevance
296 auto shifted_normalized = add_binary(operand_shape, HloOpcode::kAdd, in HandleBatchNormTraining() local299 auto tuple = HloInstruction::CreateTuple({shifted_normalized, mean, var}); in HandleBatchNormTraining()404 auto shifted_normalized = HloInstruction::CreateBinary( in HandleBatchNormInference() local424 shifted_normalized->set_sharding(sharding); in HandleBatchNormInference()427 ReplaceWithNewInstruction(batch_norm, std::move(shifted_normalized))); in HandleBatchNormInference()