/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf_with_tf2xla.cc | 291 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/ |
D | device_spec.py | 43 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
|
D | config.py | 387 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/ |
D | compile_mlir_util.cc | 222 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/ |
D | system_monitor.cc | 33 void SystemMonitor::ProcessDevicesChanged(DeviceType device_type) { in ProcessDevicesChanged() 45 void SystemMonitor::NotifyDevicesChanged(DeviceType device_type) { in NotifyDevicesChanged()
|
D | system_monitor.h | 40 virtual void OnDevicesChanged(DeviceType device_type) {} in OnDevicesChanged()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_platform_info.h | 32 explicit XlaPlatformInfo(const DeviceType device_type, in XlaPlatformInfo() 52 DeviceType device_type() const { return device_type_; } in device_type() function
|
D | partially_decluster_pass.cc | 70 DeviceType device_type(""); in FindNodesToDecluster() local 239 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice() 250 DeviceType device_type(""); in MustCompileNode() local
|
D | device_util.cc | 62 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/ |
D | device_factory.h | 136 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument 140 #define INTERNAL_REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, \ argument
|
D | device_factory.cc | 54 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()
|
D | op_kernel.cc | 229 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 …]
|
D | kernel_def.proto | 18 string device_type = 2; field
|
D | device_attributes.proto | 38 string device_type = 2; field
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | memory_types.cc | 49 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()
|
D | ring_reducer_test.cc | 103 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()
|
D | hierarchical_tree_broadcaster_test.cc | 222 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()
|
D | ring_gatherer_test.cc | 102 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/ |
D | constant_op_test.cc | 55 DeviceType device_type = DEVICE_CPU; in PersistentMemoryTrackingTest() local 58 DeviceType device_type = DEVICE_GPU; in PersistentMemoryTrackingTest() local
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | collective_param_resolver_distributed_test.cc | 139 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/ |
D | wrap_function_device_test.py | 61 def testWrapFuncDatasetDevice(self, device_type, dataset_reduce_fn): argument
|
/external/tensorflow/tensorflow/core/profiler/utils/ |
D | hardware_type_utils.cc | 112 HardwareType ParseHardwareType(absl::string_view device_type) { in ParseHardwareType()
|
/external/tensorflow/tensorflow/core/profiler/ |
D | profiler_options.proto | 24 DeviceType device_type = 6; field
|
/external/libbrillo/brillo/udev/ |
D | udev_monitor.cc | 70 const char* device_type) { in FilterAddMatchSubsystemDeviceType()
|
/external/tensorflow/tensorflow/core/common_runtime/device/ |
D | device_id_manager_test.cc | 34 DeviceType device_type("GPU"); in TEST() local
|