Home
last modified time | relevance | path

Searched refs:AutoclusteringPolicy (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cpu_device.cc63 ? XlaOpRegistry::AutoclusteringPolicy::kIfExplicitlyRequested in CreateDevices()
64 : XlaOpRegistry::AutoclusteringPolicy::kAlways; in CreateDevices()
Dxla_tpu_device.cc344 ? XlaOpRegistry::AutoclusteringPolicy::kAlways in CreateDevices()
345 : XlaOpRegistry::AutoclusteringPolicy::kIfExplicitlyRequested; in CreateDevices()
Dxla_gpu_device.cc106 XlaOpRegistry::AutoclusteringPolicy::kAlways; in CreateDevices()
Dpartially_decluster_pass.cc243 XlaOpRegistry::AutoclusteringPolicy::kAlways; in IsMustCompileDevice()
Dmark_for_compilation_pass.cc1643 XlaOpRegistry::AutoclusteringPolicy::kAlways || in ShouldCompileClusterImpl()
1645 XlaOpRegistry::AutoclusteringPolicy::kIfEnabledGlobally && in ShouldCompileClusterImpl()
Dbuild_xla_ops_pass.cc291 XlaOpRegistry::AutoclusteringPolicy::kAlways; in DeviceRequiresCompilation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.h73 enum class AutoclusteringPolicy { enum
92 AutoclusteringPolicy autoclustering_policy;
Dxla_op_registry.cc162 ? XlaOpRegistry::AutoclusteringPolicy::kIfEnabledGlobally in GetCompilationDevice()
163 : XlaOpRegistry::AutoclusteringPolicy::kIfExplicitlyRequested; in GetCompilationDevice()
170 XlaOpRegistry::AutoclusteringPolicy::kIfEnabledGlobally; in GetCompilationDevice()