Home
last modified time | relevance | path

Searched defs:device_type (Results 1 – 25 of 198) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h38 static void Register(const std::string& device_type, DeviceFactory* factory, in Register()
158 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
162 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
Ddevice_factory.cc53 bool IsDeviceFactoryEnabled(const string& device_type) { in IsDeviceFactoryEnabled()
67 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority()
78 bool DeviceFactory::IsPluggableDevice(const string& device_type) { in IsPluggableDevice()
89 void DeviceFactory::Register(const string& device_type, in Register()
113 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory()
/external/tensorflow/tensorflow/python/framework/
Ddevice_spec.py38 def _as_device_str_or_none(device_type): argument
111 device_type=None, argument
279 def device_type(self): member in DeviceSpecV2
370 def _components_to_string(job, replica, task, device_type, device_index): argument
436 def device_type(self, device_type): argument
Dconfig.py395 def list_physical_devices(device_type=None): argument
429 def list_logical_devices(device_type=None): argument
465 def get_visible_devices(device_type=None): argument
498 def set_visible_devices(devices, device_type=None): argument
Dconfig_test.py601 def testGetMemoryInfoBasic(self, device_type): argument
612 def testGetMemoryUsageSubstring(self, device_type): argument
645 def testPeakMemoryUsage(self, device_type): argument
662 def testResetMemoryStats(self, device_type): argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc430 const std::string& device_type) { in CreateDeviceMgr()
442 const std::string& device_type, in RewriteOp()
450 const std::string& device_type, bool is_module_pass) in Tf2XlaRewriter()
722 const std::string& device_type, in Tf2XlaRewritePattern()
759 explicit LegalizeTF(llvm::StringRef device_type, bool prefer_tf2xla) { in LegalizeTF()
781 void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, in PopulateLegalizeTfWithTf2XlaPatterns()
791 llvm::StringRef device_type, bool prefer_tf2xla) { in createLegalizeTfWithTf2XlaPass()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dcustom_graph_optimizer_registry.cc123 const Creator& optimizer_creator, const std::string& device_type, in RegisterPluginOptimizerOrDie()
138 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
156 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
182 for (const auto& device_type : device_types) { in GetPluginConfigs() local
/external/libchrome/base/system_monitor/
Dsystem_monitor.cc33 void SystemMonitor::ProcessDevicesChanged(DeviceType device_type) { in ProcessDevicesChanged()
45 void SystemMonitor::NotifyDevicesChanged(DeviceType device_type) { in NotifyDevicesChanged()
Dsystem_monitor.h40 virtual void OnDevicesChanged(DeviceType device_type) {} in OnDevicesChanged()
/external/cronet/base/system/
Dsystem_monitor.cc33 void SystemMonitor::ProcessDevicesChanged(DeviceType device_type) { in ProcessDevicesChanged()
45 void SystemMonitor::NotifyDevicesChanged(DeviceType device_type) { in NotifyDevicesChanged()
Dsystem_monitor.h42 virtual void OnDevicesChanged(DeviceType device_type) {} in OnDevicesChanged()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc320 bool CanInlineFunctionsPostLegalization(llvm::StringRef device_type) { in CanInlineFunctionsPostLegalization()
377 mlir::OpPassManager& pm, llvm::StringRef device_type, bool prefer_tf2xla, in CreateConvertMlirToXlaHloPipeline()
460 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
504 llvm::StringRef device_type, in BuildHloFromTfInner()
517 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
568 llvm::StringRef device_type, in BuildHloFromTf()
637 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileMlirToXlaHlo()
667 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileSerializedMlirToXlaHlo()
805 llvm::StringRef device_type, in BuildHloFromModule()
821 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileGraphToXlaHlo()
[all …]
Dattribute_utils.cc45 absl::string_view device_type(value.data(), value.size()); in HasValidCompilationAndReplicationAttributes() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.h33 const DeviceType device_type, se::Platform::Id platform_id, in XlaPlatformInfo()
52 DeviceType device_type() const { return device_type_; } in device_type() function
Dpartially_decluster_pass.cc70 DeviceType device_type(""); in FindNodesToDecluster() local
237 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
248 DeviceType device_type(""); in MustCompileNode() local
Dxla_compilation_cache.proto26 string device_type = 3; field
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc50 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
95 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
156 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
217 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
Dring_reducer_test.cc48 const DeviceType& device_type, in GetKernel()
60 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd()
71 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv()
85 const TensorShape& shape, const DeviceType& device_type, in Init()
116 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
/external/autotest/server/cros/bluetooth/
Dbluetooth_adapter_pairing_tests.py239 device_type, argument
349 def hid_reconnect_speed(self, device, device_type): argument
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op_test.cc55 DeviceType device_type = DEVICE_CPU; in PersistentMemoryTrackingTest() local
58 DeviceType device_type = DEVICE_GPU; in PersistentMemoryTrackingTest() local
/external/python/pyserial/serial/tools/
Dlist_ports_osx.py99 def get_string_property(device_type, property): argument
132 def get_int_property(device_type, property, cf_number_type): argument
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_process_state.cc46 const string& device_type, const string& platform_name) { in singleton()
60 const string& device_type, const string& platform_name) in PluggableDeviceProcessState()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc117 const string& device_type, bool nccl) { in DefineWorkers()
124 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker()
164 const string& device_type, in DefineCollectiveParams()
181 const string& device_type, in CreateCollectiveParams()
283 const string& device_type, bool nccl, in RestartWorker()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_utils.cc28 std::string device_type(type); in ValidateDeviceType() local
/external/angle/scripts/
Dupdate_extension_data.py108 def name_device(gpu, device_type): argument
124 def get_props_string(gpu, bot_os, device_os, device_type): argument

12345678