Searched refs:IsCustomCallToDnnConvolution (Results 1 – 7 of 7) sorted by relevance
144 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo);
204 if (IsCustomCallToDnnConvolution(*instruction)) { in AddBackendConstraints()
768 CHECK(IsCustomCallToDnnConvolution(*instr)); in RunOnInstruction()823 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
130 if (IsCustomCallToDnnConvolution(*instr)) { in GetRelevantConvs()
190 bool IsCustomCallToDnnConvolution(const HloInstruction& hlo) { in IsCustomCallToDnnConvolution() function213 IsCustomCallToDnnConvolution(hlo); in ImplementedAsLibraryCall()
405 if (IsCustomCallToDnnConvolution(*instr)) { in RunOnComputation()
627 if (xla::gpu::IsCustomCallToDnnConvolution(*instr)) { in EmitCustomCallOp()