Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_factory.h115 #define REGISTER_LOCAL_DEVICE_FACTORY(device_type, device_factory, ...) \ argument
119 #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()
Dmemory_types.cc48 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
93 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
150 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
205 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
Dring_reducer_test.cc98 const DeviceType& device_type, in GetKernel()
110 std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, in GetAdd()
121 std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, in GetDiv()
156 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
284 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
364 const DeviceType& device_type, in GetCollectiveReduce()
403 const DeviceType& device_type, RingReducerTest* parent) in DeviceInstance()
Dhierarchical_tree_broadcaster_test.cc213 const DeviceType& device_type, int fail_after) { in Init()
359 const DeviceType& device_type, in GetKernel()
373 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastSend()
391 const DeviceType& device_type, DeviceBase* device) { in GetCollectiveBcastRecv()
407 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
554 const DeviceType& device_type, in DeviceInstance()
Dring_gatherer_test.cc98 const DeviceType& device_type, in GetKernel()
134 const DeviceType& device_type, int num_subdivs, int fail_after) { in Init()
262 void RunTest(DataType dtype, const DeviceType& device_type, int num_workers, in RunTest()
344 const DeviceType& device_type, in GetCollectiveGather()
379 const DeviceType& device_type, RingGathererTest* parent) in DeviceInstance()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h39 explicit XlaPlatformInfo(const DeviceType device_type, in XlaPlatformInfo()
61 DeviceType device_type() const { return device_type_; } in device_type() function
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc58 DeviceType device_type(""); in FindNodesToDecluster() local
227 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
238 DeviceType device_type(""); in MustCompileNode() local
Dmark_for_compilation_pass.cc490 DeviceType device_type(""); in FindCompilationCandidates() local
714 auto is_compilable = [&](const Node* node, const DeviceType& device_type) { in Run()
931 DeviceType device_type(""); in IgnoreResourceOpForSafetyAnalysis() local
1002 DeviceType device_type(""); in AreDevicesCompatible() local
1033 DeviceType device_type(""); in ShouldCompileClusterImpl() local
Dxla_compilation_cache.h89 const DeviceType& device_type() const { return device_type_; } in device_type() function
Dxla_fusion_optimizer.cc177 DeviceType device_type(""); in Optimize() local
/external/tensorflow/tensorflow/core/kernels/
Dconstant_op_test.cc51 DeviceType device_type = DEVICE_CPU; in PersistentMemoryTrackingTest() local
54 DeviceType device_type = DEVICE_GPU; in PersistentMemoryTrackingTest() local
Dops_testutil.cc25 void OpsTestBase::SetDevice(const DeviceType& device_type, in SetDevice()
/external/autotest/server/site_tests/bluetooth_AdapterPairing/
Dbluetooth_AdapterPairing.py37 def run_once(self, host, device_type, num_iterations=1, min_pass_count=1, argument
/external/autotest/server/site_tests/bluetooth_AdapterHIDReports/
Dbluetooth_AdapterHIDReports.py43 def run_once(self, host, device_type, num_iterations=1, min_pass_count=1, argument
/external/tensorflow/tensorflow/core/framework/
Dkernel_def.proto16 string device_type = 2; field
Ddevice_attributes.proto37 string device_type = 2; field
Dkernel_def_builder.cc32 KernelDefBuilder& KernelDefBuilder::Device(const char* device_type) { in Device()
Dop_kernel.cc202 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1060 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1103 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1137 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1147 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
1182 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1200 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode() local
1257 DeviceType device_type, DeviceBase* device, Allocator* allocator, in CreateOpKernel()
1265 Status CreateOpKernel(DeviceType device_type, DeviceBase* device, in CreateOpKernel()
Dtypes.cc201 string DeviceTypeString(const DeviceType& device_type) { in DeviceTypeString()
Dmemory_types.cc76 const DeviceType& device_type, const NodeDef& ndef, in MemoryTypesForNode()
/external/u-boot/arch/x86/cpu/
Dcpu_x86.c67 const char *device_type; in cpu_x86_get_count() local
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto501 string device_type = 3; field
509 string device_type = 3; field
527 string device_type = 8; field
/external/mesa3d/src/gallium/state_trackers/clover/api/
Ddevice.cpp31 clGetDeviceIDs(cl_platform_id d_platform, cl_device_type device_type, in clGetDeviceIDs()
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc145 const string& device_type) { in DefineWorkers()
152 void DefineWorker(const string& worker_name, const string& device_type, in DefineWorker()

123