Home
last modified time | relevance | path

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

123456

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_set_test.cc52 std::vector<DeviceType> types() const { in types()
78 EXPECT_EQ(50, DeviceSet::DeviceTypeOrder(DeviceType("d1"))); in TEST_F()
79 EXPECT_EQ(51, DeviceSet::DeviceTypeOrder(DeviceType("d2"))); in TEST_F()
80 EXPECT_EQ(49, DeviceSet::DeviceTypeOrder(DeviceType("d3"))); in TEST_F()
82 EXPECT_EQ(std::vector<DeviceType>{}, types()); in TEST_F()
85 EXPECT_EQ(std::vector<DeviceType>{DeviceType("d1")}, types()); in TEST_F()
88 EXPECT_EQ(std::vector<DeviceType>{DeviceType("d1")}, types()); in TEST_F()
92 EXPECT_EQ((std::vector<DeviceType>{DeviceType("d2"), DeviceType("d1")}), in TEST_F()
97 EXPECT_EQ((std::vector<DeviceType>{ in TEST_F()
98 DeviceType("d2"), in TEST_F()
[all …]
Ddevice_set.cc61 int DeviceSet::DeviceTypeOrder(const DeviceType& d) { in DeviceTypeOrder()
65 static bool DeviceTypeComparator(const DeviceType& a, const DeviceType& b) { in DeviceTypeComparator()
77 std::vector<DeviceType> DeviceSet::PrioritizedDeviceTypeList() const { in PrioritizedDeviceTypeList()
78 std::vector<DeviceType> result; in PrioritizedDeviceTypeList()
112 std::set<DeviceType> seen; in prioritized_device_types()
114 DeviceType t(p.first->device_type()); in prioritized_device_types()
130 DeviceSet::DeviceTypeOrder(DeviceType(a.first->device_type())); in SortPrioritizedDeviceVector()
132 DeviceSet::DeviceTypeOrder(DeviceType(b.first->device_type())); in SortPrioritizedDeviceVector()
150 d, DeviceSet::DeviceTypeOrder(DeviceType(d->device_type()))); in prioritized_devices()
Dcopy_tensor.h56 Registration(DeviceType sender_device_type, DeviceType receiver_device_type, in Registration()
67 static Status Register(DeviceType sender_device_type,
68 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.cc34 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf) in RegistrationInfo()
38 DeviceType sender_device_type;
39 DeviceType receiver_device_type;
210 const DeviceType src_device_type( in ViaDMA()
212 const DeviceType dst_device_type( in ViaDMA()
214 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in ViaDMA()
215 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in ViaDMA()
296 Status CopyTensor::Register(DeviceType sender_device_type, in Register()
297 DeviceType receiver_device_type, in Register()
Dcollective_rma_local.cc123 const DeviceType src_device_type( in MemCpyAsync()
125 const DeviceType dst_device_type( in MemCpyAsync()
127 const bool non_cpu_src = src_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync()
128 const bool non_cpu_dst = dst_device_type != DeviceType(DEVICE_CPU); in MemCpyAsync()
Ddevice_set.h68 std::vector<DeviceType> PrioritizedDeviceTypeList() const;
85 static int DeviceTypeOrder(const DeviceType& d);
Dmemory_types.cc49 const DeviceType& device_type, const Graph* g, in ProcessMemoryTypes()
94 Status ValidateMemoryTypes(const DeviceType& device_type, const Graph* g) { in ValidateMemoryTypes()
151 Status EnsureMemoryTypes(const DeviceType& device_type, in EnsureMemoryTypes()
212 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput()
/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/frontends/nine/
Dadapter9.h89 D3DDEVTYPE DeviceType,
97 D3DDEVTYPE DeviceType,
105 D3DDEVTYPE DeviceType,
112 D3DDEVTYPE DeviceType,
118 D3DDEVTYPE DeviceType,
124 D3DDEVTYPE DeviceType,
135 D3DDEVTYPE DeviceType,
Dadapter9.c280 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDeviceFormat() argument
296 nine_D3DDEVTYPE_to_str(DeviceType), d3dformat_to_string(AdapterFormat)); in NineAdapter9_CheckDeviceFormat()
304 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDeviceFormat()
408 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDeviceMultiSampleType() argument
420 "pQualityLevels=%p\n", This, nine_D3DDEVTYPE_to_str(DeviceType), in NineAdapter9_CheckDeviceMultiSampleType()
430 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDeviceMultiSampleType()
470 D3DDEVTYPE DeviceType, in NineAdapter9_CheckDepthStencilMatch() argument
481 nine_D3DDEVTYPE_to_str(DeviceType), d3dformat_to_string(AdapterFormat), in NineAdapter9_CheckDepthStencilMatch()
491 hr = NineAdapter9_GetScreen(This, DeviceType, &screen); in NineAdapter9_CheckDepthStencilMatch()
516 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/profiler/internal/
Dprofiler_interface.h25 enum class DeviceType { enum
37 DeviceType device_type = DeviceType::kUnspecified;
72 virtual DeviceType GetDeviceType() = 0;
/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 <<()
203 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);
108 typedef gtl::InlinedVector<DeviceType, 4> DeviceTypeVector;
109 typedef gtl::InlinedVector<std::pair<DeviceType, int32>, 4>
114 string DeviceTypeString(const DeviceType& device_type);
Dop_kernel_test.cc131 static std::vector<DeviceType> DeviceTypes() { in DeviceTypes()
132 return {DeviceType(DEVICE_GPU), DeviceType(DEVICE_CPU)}; in DeviceTypes()
163 void ExpectSuccess(const string& op_type, DeviceType device_type, in ExpectSuccess()
178 void ExpectFailure(const string& ascii_node_def, DeviceType device_type, in ExpectFailure()
219 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
236 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
237 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[1].first); in TEST_F()
262 EXPECT_EQ(DeviceType(DEVICE_CPU), devs[0].first); in TEST_F()
271 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
287 EXPECT_EQ(DeviceType(DEVICE_GPU), devs[0].first); in TEST_F()
[all …]
Dop_kernel.cc126 expensive_ = context->device_type() != DeviceType(DEVICE_GPU) && in OpKernel()
127 context->device_type() != DeviceType(DEVICE_SYCL); in OpKernel()
212 DeviceType device_type, DeviceBase* device, Allocator* allocator, in OpKernelConstruction()
1233 static string Key(StringPiece op_type, const DeviceType& device_type, in Key()
1247 Key(kernel_def->op(), DeviceType(kernel_def->device_type()), in InitInternal()
1280 const DeviceType& device_type, StringPiece node_name, in FindKernelRegistration()
1358 Status FindKernelRegistration(const DeviceType& device_type, in FindKernelRegistration()
1370 bool KernelDefAvailable(const DeviceType& device_type, in KernelDefAvailable()
1381 const DeviceType& device_type, StringPiece node_name, in FindKernelDef()
1412 Status FindKernelDef(const DeviceType& device_type, const NodeDef& node_def, in FindKernelDef()
[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()
/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util.h134 const DeviceType& GetDeviceTypeFor(DeviceId device) const { in GetDeviceTypeFor()
138 using DeviceTypeConstRef = std::reference_wrapper<const DeviceType>;
155 std::vector<std::unique_ptr<DeviceType>> id_to_device_type_;
164 Status DeviceNameToDeviceType(const string& device, DeviceType* device_type);
Dxla_device.h62 const DeviceType& device_type,
71 const DeviceType& jit_device_type() const;
81 const DeviceType device_type_;
215 const DeviceType jit_device_name_;
Dxla_compilation_cache.h47 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
90 const DeviceType& device_type() const { return device_type_; } in device_type()
141 const DeviceType device_type_;
Ddevice_util.cc61 id_to_device_type_.push_back(absl::make_unique<DeviceType>("")); in GetIdFor()
62 DeviceType* device_type = id_to_device_type_.back().get(); in GetIdFor()
91 Status DeviceNameToDeviceType(const string& device, DeviceType* device_type) { in DeviceNameToDeviceType()
96 *device_type = DeviceType(parsed.type); in DeviceNameToDeviceType()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_test.cc93 std::make_pair(DeviceType(DEVICE_GPU), 20), in TEST_F()
94 std::make_pair(DeviceType(DEVICE_CPU), 10), in TEST_F()
139 std::make_pair(DeviceType(DEVICE_GPU), 20), in TEST_F()
140 std::make_pair(DeviceType(DEVICE_CPU), 10), in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.h40 explicit XlaPlatformInfo(const DeviceType device_type, in XlaPlatformInfo()
60 DeviceType device_type() const { return device_type_; } in device_type()
74 DeviceType device_type_;
/external/pdfium/core/fxge/win32/
Dwin32_int.h77 CGdiDeviceDriver(HDC hDC, DeviceType device_type);
81 DeviceType GetDeviceType() const override;
132 const DeviceType m_DeviceType;
231 DeviceType GetDeviceType() const override;
297 DeviceType GetDeviceType() const override;

123456