Home
last modified time | relevance | path

Searched defs:custom_call_handler (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference.cc75 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler, in DynamicDimensionInferenceVisitor()
87 DynamicDimensionInference::CustomCallInferenceHandler in Run()
1946 HloModule* module, CustomCallInferenceHandler custom_call_handler, in Run()
1971 HloModule* module, CustomCallInferenceHandler custom_call_handler, in DynamicDimensionInference()
H A Ddynamic_padder.h65 DynamicDimensionInference::CustomCallInferenceHandler custom_call_handler = member