Searched defs:DeviceRegistration (Results 1 – 2 of 2) sorted by relevance
87 struct DeviceRegistration { struct89 string compilation_device_name;92 AutoclusteringPolicy autoclustering_policy;96 bool cluster_resource_variable_ops_unsafely = false;99 bool cluster_stack_ops = false;102 bool cluster_tensor_array_ops = false;107 bool cluster_stateful_rng_ops = false;113 bool cluster_control_trigger = false;117 bool elide_assert_and_checknumerics = false;120 bool cluster_variant_ops = false;[all …]
123 using DeviceRegistration = const XlaOpRegistry::DeviceRegistration; variable