Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.h481 int64_t FindRotateRightPattern(const HloInstruction* concat,
H A Dspmd_partitioner_util.cc1811 int64_t FindRotateRightPattern(const HloInstruction* concat, in FindRotateRightPattern() function
H A Dspmd_partitioner.cc4737 const int64_t amount = FindRotateRightPattern(hlo, lhs, rhs); in PreprocessHlos()