Searched defs:custom_call_handler (Results 1 – 1 of 1) sorted by relevance
74 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler) in DynamicDimensionInferenceVisitor()84 DynamicDimensionInference::CustomCallInferenceHandler in Run()1641 HloModule* module, CustomCallInferenceHandler custom_call_handler) { in Run()1663 HloModule* module, CustomCallInferenceHandler custom_call_handler) in DynamicDimensionInference()