Searched refs:parameter_use (Results 1 – 1 of 1) sorted by relevance
906 for (const HloUse& parameter_use : parameter_value->uses()) { in IsUseAllowedInAlternateMemory() local907 int64 use_time = instruction_schedule.at(parameter_use.instruction); in IsUseAllowedInAlternateMemory()908 if (parameter_use.instruction->opcode() != HloOpcode::kGetTupleElement && in IsUseAllowedInAlternateMemory()909 parameter_use.instruction->opcode() != HloOpcode::kTuple && in IsUseAllowedInAlternateMemory()910 parameter_use.instruction->opcode() != HloOpcode::kBitcast && in IsUseAllowedInAlternateMemory()963 for (const HloUse& parameter_use : parameter_value->uses()) { in IsUseAllowedInAlternateMemory() local964 if (parameter_use.instruction->parent() == called_computation && in IsUseAllowedInAlternateMemory()965 parameter_use.instruction->opcode() != in IsUseAllowedInAlternateMemory()967 parameter_use.instruction->opcode() != HloOpcode::kTuple && in IsUseAllowedInAlternateMemory()968 parameter_use.instruction->opcode() != HloOpcode::kBitcast) { in IsUseAllowedInAlternateMemory()[all …]