Home
last modified time | relevance | path

Searched refs:MlirBridgeRolloutPolicy (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_bridge_pass.cc96 MlirBridgeRolloutPolicy policy = GetMlirBridgeRolloutPolicy( in GetPassState()
99 case MlirBridgeRolloutPolicy::kEnabledByUser: in GetPassState()
101 case MlirBridgeRolloutPolicy::kEnabledAfterGraphAnalysis: in GetPassState()
103 case MlirBridgeRolloutPolicy::kDisabledByUser: in GetPassState()
104 case MlirBridgeRolloutPolicy::kDisabledAfterGraphAnalysis: in GetPassState()
151 MlirBridgeRolloutPolicy policy = GetMlirBridgeRolloutPolicy( in IsEnabled()
153 return policy == MlirBridgeRolloutPolicy::kEnabledByUser; in IsEnabled()
Dxla_compiler.cc807 MlirBridgeRolloutPolicy policy = GetMlirBridgeRolloutPolicy( in CompileFunction()
810 if (policy == MlirBridgeRolloutPolicy::kEnabledByUser) { in CompileFunction()
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_bridge_rollout_policy.cc53 MlirBridgeRolloutPolicy GetMlirBridgeRolloutPolicy( in GetMlirBridgeRolloutPolicy()
58 return MlirBridgeRolloutPolicy::kEnabledByUser; in GetMlirBridgeRolloutPolicy()
60 return MlirBridgeRolloutPolicy::kDisabledByUser; in GetMlirBridgeRolloutPolicy()
64 return MlirBridgeRolloutPolicy::kDisabledAfterGraphAnalysis; in GetMlirBridgeRolloutPolicy()
Dmlir_bridge_rollout_policy.h25 enum class MlirBridgeRolloutPolicy { enum
51 MlirBridgeRolloutPolicy GetMlirBridgeRolloutPolicy(
/external/tensorflow/tensorflow/compiler/jit/
Dxla_kernel_creator.cc113 MlirBridgeRolloutPolicy policy = GetMlirBridgeRolloutPolicy( in CreateXlaKernel()
115 if (policy != MlirBridgeRolloutPolicy::kEnabledByUser) { in CreateXlaKernel()
Dxla_compilation_cache.cc292 MlirBridgeRolloutPolicy::kEnabledByUser && in CompileSingleOp()