Home
last modified time | relevance | path

Searched refs:DeviceType (Results 1 – 25 of 102) sorted by relevance

12345

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_set_test.cc48 std::vector<DeviceType> types() const { in types()
71 EXPECT_EQ(50, DeviceSet::DeviceTypeOrder(DeviceType("d1"))); in TEST_F()
72 EXPECT_EQ(51, DeviceSet::DeviceTypeOrder(DeviceType("d2"))); in TEST_F()
73 EXPECT_EQ(49, DeviceSet::DeviceTypeOrder(DeviceType("d3"))); in TEST_F()
75 EXPECT_EQ(std::vector<DeviceType>{}, types()); in TEST_F()
78 EXPECT_EQ(std::vector<DeviceType>{DeviceType("d1")}, types()); in TEST_F()
81 EXPECT_EQ(std::vector<DeviceType>{DeviceType("d1")}, types()); in TEST_F()
85 EXPECT_EQ((std::vector<DeviceType>{DeviceType("d2"), DeviceType("d1")}), in TEST_F()
90 EXPECT_EQ((std::vector<DeviceType>{ in TEST_F()
91 DeviceType("d2"), in TEST_F()
[all …]
Ddevice_set.cc58 int DeviceSet::DeviceTypeOrder(const DeviceType& d) { in DeviceTypeOrder()
62 static bool DeviceTypeComparator(const DeviceType& a, const DeviceType& b) { in DeviceTypeComparator()
74 std::vector<DeviceType> DeviceSet::PrioritizedDeviceTypeList() const { in PrioritizedDeviceTypeList()
75 std::vector<DeviceType> result; in PrioritizedDeviceTypeList()
Dcopy_tensor.h55 Registration(DeviceType sender_device_type, DeviceType receiver_device_type, in Registration()
66 static Status Register(DeviceType sender_device_type,
67 DeviceType receiver_device_type,
Dmemory_types.h27 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g);
38 Status EnsureMemoryTypes(const DeviceType& device_type,
43 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g,
Dcopy_tensor.cc33 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf) in RegistrationInfo()
37 DeviceType sender_device_type;
38 DeviceType receiver_device_type;
258 const DeviceType src_device_type( in ViaDMA()
260 const DeviceType dst_device_type( in ViaDMA()
262 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in ViaDMA()
263 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in ViaDMA()
348 Status CopyTensor::Register(DeviceType sender_device_type, in Register()
349 DeviceType receiver_device_type, in Register()
Dcollective_rma_local.cc94 const DeviceType src_device_type( in MemCpyAsync()
96 const DeviceType dst_device_type( in MemCpyAsync()
98 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync()
99 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync()
Ddevice_set.h65 std::vector<DeviceType> PrioritizedDeviceTypeList() const;
71 static int DeviceTypeOrder(const DeviceType& d);
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()
531 DeviceType device_type_;
542 DeviceType device_type_;
562 cp.group.device_type = DeviceType("GPU"); in SetUpCollectiveParams()
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()
536 cp->group.device_type = DeviceType("GPU"); in PrepColParamsForSubdivPermsTest()
554 const DeviceType& device_type, in DeviceInstance()
700 DeviceType device_type_ = DEVICE_CPU;
711 DeviceType device_type_;
788 cp.group.device_type = DeviceType("GPU"); in TEST_F()
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()
503 DeviceType device_type_;
515 DeviceType device_type_;
535 cp.group.device_type = DeviceType("GPU"); in SetUpCollectiveParams()
/external/mesa3d/include/d3dadapter/
Dd3dadapter9.h42 …HRESULT (WINAPI *CheckDeviceFormat)(ID3DAdapter9 *This, D3DDEVTYPE DeviceType, D3DFORMAT AdapterFo…
43 …HRESULT (WINAPI *CheckDeviceMultiSampleType)(ID3DAdapter9 *This, D3DDEVTYPE DeviceType, D3DFORMAT …
44 …HRESULT (WINAPI *CheckDepthStencilMatch)(ID3DAdapter9 *This, D3DDEVTYPE DeviceType, D3DFORMAT Adap…
45 …HRESULT (WINAPI *CheckDeviceFormatConversion)(ID3DAdapter9 *This, D3DDEVTYPE DeviceType, D3DFORMAT…
46 HRESULT (WINAPI *GetDeviceCaps)(ID3DAdapter9 *This, D3DDEVTYPE DeviceType, D3DCAPS9 *pCaps);
47 …HRESULT (WINAPI *CreateDevice)(ID3DAdapter9 *This, UINT RealAdapter, D3DDEVTYPE DeviceType, HWND h…
48 …HRESULT (WINAPI *CreateDeviceEx)(ID3DAdapter9 *This, UINT RealAdapter, D3DDEVTYPE DeviceType, HWND…
77 …HRESULT WINAPI CheckDeviceFormat(D3DDEVTYPE DeviceType, D3DFORMAT AdapterFormat, DWORD Usage, D3DR…
78 …HRESULT WINAPI CheckDeviceMultiSampleType(D3DDEVTYPE DeviceType, D3DFORMAT SurfaceFormat, BOOL Win…
79 …HRESULT WINAPI CheckDepthStencilMatch(D3DDEVTYPE DeviceType, D3DFORMAT AdapterFormat, D3DFORMAT Re…
[all …]
/external/mesa3d/src/gallium/state_trackers/nine/
Dadapter9.h87 D3DDEVTYPE DeviceType,
95 D3DDEVTYPE DeviceType,
103 D3DDEVTYPE DeviceType,
110 D3DDEVTYPE DeviceType,
116 D3DDEVTYPE DeviceType,
122 D3DDEVTYPE DeviceType,
133 D3DDEVTYPE DeviceType,
Dadapter9.c250 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDeviceFormat() argument
266 nine_D3DDEVTYPE_to_str(DeviceType), d3dformat_to_string(AdapterFormat)); in NineAdapter9_CheckDeviceFormat()
272 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDeviceFormat()
376 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDeviceMultiSampleType() argument
388 "pQualityLevels=%p\n", This, nine_D3DDEVTYPE_to_str(DeviceType), in NineAdapter9_CheckDeviceMultiSampleType()
398 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDeviceMultiSampleType()
438 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDepthStencilMatch() argument
449 nine_D3DDEVTYPE_to_str(DeviceType), d3dformat_to_string(AdapterFormat), in NineAdapter9_CheckDepthStencilMatch()
456 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDepthStencilMatch()
481 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDeviceFormatConversion() argument
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDevice.h27 template <typename ExpressionType, typename DeviceType> class TensorDevice {
29 …TensorDevice(const DeviceType& device, ExpressionType& expression) : m_device(device), m_expressio… in TensorDevice()
35 internal::TensorExecutor<const Assign, DeviceType>::run(assign, m_device);
46 internal::TensorExecutor<const Assign, DeviceType>::run(assign, m_device);
57 internal::TensorExecutor<const Assign, DeviceType>::run(assign, m_device);
62 const DeviceType& m_device;
/external/tensorflow/tensorflow/core/framework/
Dtypes.cc25 bool DeviceType::operator<(const DeviceType& other) const { in operator <()
29 bool DeviceType::operator==(const DeviceType& other) const { in operator ==()
33 std::ostream& operator<<(std::ostream& os, const DeviceType& d) { in operator <<()
201 string DeviceTypeString(const DeviceType& device_type) { in DeviceTypeString()
Dtypes.h54 class DeviceType {
56 DeviceType(const char* type) // NOLINT(runtime/explicit) in DeviceType() function
59 explicit DeviceType(StringPiece type) : type_(type.data(), type.size()) {} in DeviceType() function
64 bool operator<(const DeviceType& other) const;
65 bool operator==(const DeviceType& other) const;
66 bool operator!=(const DeviceType& other) const { return !(*this == other); }
71 std::ostream& operator<<(std::ostream& os, const DeviceType& d);
106 typedef gtl::InlinedVector<DeviceType, 4> DeviceTypeVector;
107 typedef gtl::InlinedVector<std::pair<DeviceType, int32>, 4>
112 string DeviceTypeString(const DeviceType& device_type);
Dop_kernel_test.cc126 static std::vector<DeviceType> DeviceTypes() { in DeviceTypes()
127 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
156 void ExpectSuccess(const string& op_type, DeviceType device_type, in ExpectSuccess()
171 void ExpectFailure(const string& ascii_node_def, DeviceType device_type, in ExpectFailure()
212 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
223 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
232 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
248 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
249 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[1].first); in TEST_F()
258 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
[all …]
Dtypes_test.cc27 EXPECT_EQ("CPU", DeviceTypeString(DeviceType(DEVICE_CPU))); in TEST()
28 EXPECT_EQ("GPU", DeviceTypeString(DeviceType(DEVICE_GPU))); in TEST()
29 EXPECT_EQ("SYCL", DeviceTypeString(DeviceType(DEVICE_SYCL))); in TEST()
Dop_kernel.cc118 expensive_ = context->device_type() != DeviceType(DEVICE_GPU) && in OpKernel()
119 context->device_type() != DeviceType(DEVICE_SYCL); in OpKernel()
202 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1060 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1074 Key(kernel_def->op(), DeviceType(kernel_def->device_type()), in InitInternal()
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()
1173 const std::vector<DeviceType>& prioritized_types, const NodeDef& def, in SupportedDeviceTypesForNode()
1182 for (const DeviceType& device_type : prioritized_types) { in SupportedDeviceTypesForNode()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h62 const DeviceType& device_type,
71 const DeviceType& jit_device_type() const;
81 const DeviceType device_type_;
202 const DeviceType jit_device_name_;
Dmark_for_compilation_pass.cc117 bool HasXLAKernel(const Node& node, const DeviceType& jit_device_type) { in HasXLAKernel()
167 const DeviceType& jit_device_type,
175 const DeviceType& jit_device_type, in IsCompilableWhile()
220 const DeviceType& jit_device_type, in IsCompilableCall()
450 const std::function<bool(const Node*, const DeviceType&)>& is_compilable_fn, in FindCompilationCandidates()
490 DeviceType device_type(""); in FindCompilationCandidates()
504 DeviceType jit_device_type(registration->compilation_device_name); in FindCompilationCandidates()
658 DeviceType jit_device_type(registration->compilation_device_name); in IsCompilable()
714 auto is_compilable = [&](const Node* node, const DeviceType& device_type) { in Run()
931 DeviceType device_type(""); in IgnoreResourceOpForSafetyAnalysis()
[all …]
Dxla_compilation_cache.h46 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
89 const DeviceType& device_type() const { return device_type_; } in device_type()
139 const DeviceType device_type_;
Dpartially_decluster_pass.cc58 DeviceType device_type(""); in FindNodesToDecluster()
79 DeviceType dst_device_type(""); in FindNodesToDecluster()
227 bool IsMustCompileDevice(const DeviceType& device_type) { in IsMustCompileDevice()
238 DeviceType device_type(""); in MustCompileNode()
/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()
75 DeviceType device_type_;

12345