Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc291 const std::string& device_type) { in CreateDeviceMgr()
303 const std::string& device_type) { in RewriteOp()
310 const std::string& device_type) in Tf2XlaRewriter()
575 explicit Tf2XlaRewritePattern(const std::string& device_type) in Tf2XlaRewritePattern()
593 explicit LegalizeTF(llvm::StringRef device_type) { in LegalizeTF()
621 void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, in PopulateLegalizeTfWithTf2XlaPatterns()
627 llvm::StringRef device_type) { in createLegalizeTfWithTf2XlaPass()
/external/tensorflow/tensorflow/python/framework/
Ddevice_spec.py43 def _as_device_str_or_none(device_type): argument
112 def __init__(self, job=None, replica=None, task=None, device_type=None, argument
278 def device_type(self): member in DeviceSpecV2
356 def _components_to_string(job, replica, task, device_type, device_index): argument
422 def device_type(self, device_type): member in DeviceSpecV1
Dconfig.py387 def list_physical_devices(device_type=None): argument
422 def list_logical_devices(device_type=None): argument
459 def get_visible_devices(device_type=None): argument
493 def set_visible_devices(devices, device_type=None): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc222 bool CanInlineFunctionsPostLegalization(llvm::StringRef device_type) { in CanInlineFunctionsPostLegalization()
279 mlir::OpPassManager& pm, llvm::StringRef device_type, in CreateConvertMlirToXlaHloPipeline()
341 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
378 llvm::StringRef device_type, in BuildHloFromTfInner()
389 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
425 llvm::StringRef device_type, in BuildHloFromTf()
469 llvm::StringRef device_type, bool use_tuple_args, bool use_return_tuple, in CompileMlirToXlaHlo()
492 llvm::StringRef device_type, bool use_tuple_args, in CompileSerializedMlirToXlaHlo()
620 llvm::StringRef device_type, in BuildHloFromModule()
633 llvm::StringRef device_type, bool use_tuple_args, bool use_return_tuple, in CompileGraphToXlaHlo()
[all …]
/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/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.h32 explicit XlaPlatformInfo(const DeviceType device_type, in XlaPlatformInfo()
52 DeviceType device_type() const { return device_type_; } in device_type() function
Dpartially_decluster_pass.cc70 DeviceType device_type(""); in FindNodesToDecluster() local
239 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
250 DeviceType device_type(""); in MustCompileNode() local
Ddevice_util.cc62 DeviceType* device_type = id_to_device_type_.back().get(); in GetIdFor() local
91 Status DeviceNameToDeviceType(const string& device, DeviceType* device_type) { in DeviceNameToDeviceType()
/external/tensorflow/tensorflow/core/framework/
Ddevice_factory.h136 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
140 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
Ddevice_factory.cc54 int32 DeviceFactory::DevicePriority(const string& device_type) { in DevicePriority()
66 void DeviceFactory::Register(const string& device_type, DeviceFactory* factory, in Register()
84 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory()
Dop_kernel.cc229 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1248 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1306 const DeviceType& device_type, StringPiece node_name, in FindKernelRegistration()
1384 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1396 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1407 const DeviceType& device_type, StringPiece node_name, in FindKernelDef()
1445 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
1465 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1496 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1517 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
[all …]
Dkernel_def.proto18 string device_type = 2; field
Ddevice_attributes.proto38 string device_type = 2; field
/external/tensorflow/tensorflow/core/common_runtime/
Dmemory_types.cc49 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
93 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
154 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
215 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
Dring_reducer_test.cc103 const DeviceType& device_type, in GetKernel()
115 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd()
126 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv()
163 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
297 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
375 const DeviceType& device_type, in GetCollectiveReduce()
415 const DeviceType& device_type, RingReducerTest* parent) in DeviceInstance()
Dhierarchical_tree_broadcaster_test.cc222 const DeviceType& device_type, int fail_after) { in Init()
373 const DeviceType& device_type, in GetKernel()
387 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastSend()
405 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastRecv()
421 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
560 const DeviceType& device_type, in DeviceInstance()
Dring_gatherer_test.cc102 const DeviceType& device_type, in GetKernel()
140 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
273 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
353 const DeviceType& device_type, in GetCollectiveGather()
389 const DeviceType& device_type, RingGathererTest* parent) in DeviceInstance()
/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/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc139 const string& device_type, bool nccl) { in DefineWorkers()
146 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker()
179 const string& device_type) { in DefineCollectiveParams()
192 const string& device_type) { in CreateCollectiveParams()
290 const string& device_type, bool nccl) { in RestartWorker()
/external/tensorflow/tensorflow/python/eager/
Dwrap_function_device_test.py61 def testWrapFuncDatasetDevice(self, device_type, dataset_reduce_fn): argument
/external/tensorflow/tensorflow/core/profiler/utils/
Dhardware_type_utils.cc112 HardwareType ParseHardwareType(absl::string_view device_type) { in ParseHardwareType()
/external/tensorflow/tensorflow/core/profiler/
Dprofiler_options.proto24 DeviceType device_type = 6; field
/external/libbrillo/brillo/udev/
Dudev_monitor.cc70 const char* device_type) { in FilterAddMatchSubsystemDeviceType()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_manager_test.cc34 DeviceType device_type("GPU"); in TEST() local

123456