Home
last modified time | relevance | path

Searched refs:NumUnnestedReductions (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_fusible.cc366 static int64_t NumUnnestedReductions(const HloInstruction& instr, in NumUnnestedReductions() function
420 if (NumUnnestedReductions(instr1, cache) + in FusionFitsInBudget()
421 NumUnnestedReductions(instr2, cache) > in FusionFitsInBudget()