Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.h29 std::unique_ptr<HloInstruction> CreateCustomCallSPMDInternal_RotateRight(
H A Dcustom_call_handler.cc369 std::unique_ptr<HloInstruction> CreateCustomCallSPMDInternal_RotateRight( in CreateCustomCallSPMDInternal_RotateRight() function
H A Dspmd_partitioner.cc4743 CreateCustomCallSPMDInternal_RotateRight(to_rotate, dim, amount)); in PreprocessHlos()
4795 CreateCustomCallSPMDInternal_RotateRight(pad, dim, in PreprocessHlos()
4828 CreateCustomCallSPMDInternal_RotateRight(pad, dim, in PreprocessHlos()