Home
last modified time | relevance | path

Searched refs:ShapeCheckMode (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference.h40 enum ShapeCheckMode { enum
60 ShapeCheckMode shape_check_mode = ShapeCheckMode::kIgnore,
105 ShapeCheckMode shape_check_mode, AssertionGenerator assertion_generator);
173 ShapeCheckMode shape_check_mode_;
Ddynamic_padder.h78 DynamicDimensionInference::ShapeCheckMode shape_check_mode =
79 DynamicDimensionInference::ShapeCheckMode::kIgnore;
Ddynamic_dimension_inference.cc76 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()
Ddynamic_dimension_inference_test.cc50 DynamicDimensionInference::ShapeCheckMode shape_check_mode = in RunInference()
51 DynamicDimensionInference::ShapeCheckMode::kIgnore, in RunInference()
1372 /*handler=*/nullptr, DynamicDimensionInference::ShapeCheckMode::kRuntime, in TEST_F()
Ddynamic_padder_test.cc2128 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in TEST_F()
2150 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc475 DynamicDimensionInference::ShapeCheckMode::kIgnore; in OptimizeHloModule()
479 DynamicDimensionInference::ShapeCheckMode::kRuntime; in OptimizeHloModule()
493 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in OptimizeHloModule()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc527 DynamicDimensionInference::ShapeCheckMode::kCompileTime; in RunHloPassesThroughLayoutAssn()