Searched refs:ShapeCheckMode (Results 1 – 7 of 7) sorted by relevance
40 enum ShapeCheckMode { enum60 ShapeCheckMode shape_check_mode = ShapeCheckMode::kIgnore,105 ShapeCheckMode shape_check_mode, AssertionGenerator assertion_generator);173 ShapeCheckMode shape_check_mode_;
78 DynamicDimensionInference::ShapeCheckMode shape_check_mode =79 DynamicDimensionInference::ShapeCheckMode::kIgnore;
76 DynamicDimensionInference::ShapeCheckMode shape_check_mode) in DynamicDimensionInferenceVisitor()89 DynamicDimensionInference::ShapeCheckMode shape_check_mode = in Run()90 DynamicDimensionInference::ShapeCheckMode::kIgnore, in Run()226 DynamicDimensionInference::ShapeCheckMode shape_check_mode_;1947 ShapeCheckMode shape_check_mode, in Run()1972 ShapeCheckMode shape_check_mode, AssertionGenerator assertion_generator) in DynamicDimensionInference()
50 DynamicDimensionInference::ShapeCheckMode shape_check_mode = in RunInference()51 DynamicDimensionInference::ShapeCheckMode::kIgnore, in RunInference()1372 /*handler=*/nullptr, DynamicDimensionInference::ShapeCheckMode::kRuntime, in TEST_F()
2128 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in TEST_F()2150 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in TEST_F()
475 DynamicDimensionInference::ShapeCheckMode::kIgnore; in OptimizeHloModule()479 DynamicDimensionInference::ShapeCheckMode::kRuntime; in OptimizeHloModule()493 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in OptimizeHloModule()
527 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in RunHloPassesThroughLayoutAssn()