Home
last modified time | relevance | path

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

1234567

/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
366 def _components_to_string(job, replica, task, device_type, device_index): argument
432 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.py596 def testGetMemoryInfoBasic(self, device_type): argument
607 def testGetMemoryUsageSubstring(self, device_type): argument
640 def testPeakMemoryUsage(self, device_type): argument
657 def testResetMemoryStats(self, device_type): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc224 bool CanInlineFunctionsPostLegalization(llvm::StringRef device_type) { in CanInlineFunctionsPostLegalization()
281 mlir::OpPassManager& pm, llvm::StringRef device_type, bool prefer_tf2xla, in CreateConvertMlirToXlaHloPipeline()
359 Status LegalizeToHlo(mlir::ModuleOp module_op, llvm::StringRef device_type, in LegalizeToHlo()
397 llvm::StringRef device_type, in BuildHloFromTfInner()
409 mlir::ModuleOp module_op, llvm::StringRef device_type, in ConvertMLIRToXlaComputation()
445 llvm::StringRef device_type, in BuildHloFromTf()
489 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileMlirToXlaHlo()
518 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileSerializedMlirToXlaHlo()
656 llvm::StringRef device_type, in BuildHloFromModule()
669 llvm::StringRef device_type, bool use_tuple_args, bool analyse_graph, in CompileGraphToXlaHlo()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc424 const std::string& device_type) { in CreateDeviceMgr()
436 const std::string& device_type) { in RewriteOp()
443 const std::string& device_type) in Tf2XlaRewriter()
715 const std::string& device_type, in Tf2XlaRewritePattern()
745 explicit LegalizeTF(llvm::StringRef device_type, bool prefer_tf2xla) { in LegalizeTF()
790 void PopulateLegalizeTfWithTf2XlaPatterns(llvm::StringRef device_type, in PopulateLegalizeTfWithTf2XlaPatterns()
799 llvm::StringRef device_type, bool prefer_tf2xla) { in createLegalizeTfWithTf2XlaPass()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dcustom_graph_optimizer_registry.cc121 const Creator& optimizer_creator, const std::string& device_type, in RegisterPluginOptimizerOrDie()
136 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
154 for (const auto& device_type : device_types) { in PrintPluginConfigsIfConflict() local
180 for (const auto& device_type : device_types) { in GetPluginConfigs() local
/external/tensorflow/tensorflow/core/framework/
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, DeviceFactory* factory, in Register()
113 DeviceFactory* DeviceFactory::GetFactory(const string& device_type) { in GetFactory()
Ddevice_factory.h150 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
154 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
Dop_kernel.cc222 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1178 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1236 const DeviceType& device_type, StringPiece node_name, in FindKernelRegistration()
1314 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1326 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1337 const DeviceType& device_type, StringPiece node_name, in FindKernelDef()
1375 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
1395 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1426 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1447 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
[all …]
Dkernel_def.proto18 string device_type = 2; field
/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.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
239 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
250 DeviceType device_type(""); in MustCompileNode() local
Ddevice_util.cc61 DeviceType* device_type = id_to_device_type_.back().get(); in GetIdFor() local
90 Status DeviceNameToDeviceType(const string& device, DeviceType* device_type) { in DeviceNameToDeviceType()
/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()
Dpermuter_test.cc52 const TensorShape& shape, const DeviceType& device_type, in Init()
85 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
/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()
282 const string& device_type, bool nccl, in RestartWorker()
/external/angle/scripts/
Dupdate_extension_data.py106 def name_device(gpu, device_type): argument
122 def get_props_string(gpu, bot_os, device_os, device_type): argument
/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()

1234567