Searched refs:bf16_called_comps (Results 1 – 1 of 1) sorted by relevance
72 HloInstruction* hlo, absl::Span<HloComputation* const> bf16_called_comps);121 HloInstruction* hlo, absl::Span<HloComputation* const> bf16_called_comps) { in ConvertCalledComputations() argument123 for (auto& comp : bf16_called_comps) { in ConvertCalledComputations()213 std::vector<HloComputation*> bf16_called_comps; in HandleMultipleOutputs() local231 bf16_called_comps.push_back(comp); in HandleMultipleOutputs()266 return ConvertCalledComputations(hlo, bf16_called_comps); in HandleMultipleOutputs()287 std::vector<HloComputation*> bf16_called_comps; in HandleInstruction() local305 bf16_called_comps.push_back(comp); in HandleInstruction()371 return ConvertCalledComputations(hlo, bf16_called_comps); in HandleInstruction()