Searched refs:CustomCallInferenceHandler (Results 1 – 4 of 4) sorted by relevance
50 using CustomCallInferenceHandler = variable59 CustomCallInferenceHandler custom_call_handler = nullptr,104 HloModule* module, CustomCallInferenceHandler custom_call_handler,170 CustomCallInferenceHandler custom_call_handler_;
65 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler =
75 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler, in DynamicDimensionInferenceVisitor()87 DynamicDimensionInference::CustomCallInferenceHandler in Run()223 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler_;1946 HloModule* module, CustomCallInferenceHandler custom_call_handler, in Run()1971 HloModule* module, CustomCallInferenceHandler custom_call_handler, in DynamicDimensionInference()
49 DynamicDimensionInference::CustomCallInferenceHandler handler = nullptr, in RunInference()