Home
last modified time | relevance | path

Searched defs:assertion_generator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_padder.h74 DynamicDimensionInference::AssertionGenerator assertion_generator; member
H A Ddynamic_dimension_inference.cc91 const DynamicDimensionInference::AssertionGenerator& in Run()
1948 const AssertionGenerator& assertion_generator) { in Run()
1972 ShapeCheckMode shape_check_mode, AssertionGenerator assertion_generator) in DynamicDimensionInference()
H A Ddynamic_dimension_inference_test.cc52 const DynamicDimensionInference::AssertionGenerator& assertion_generator = in RunInference()