Home
last modified time | relevance | path

Searched refs:GetDevice (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc126 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in MakeReductionTestCase()
175 auto* device = GetDevice(num_ranks_per_node, node, i); in MakeGatherTestCase()
220 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in MakeBroadcastTestCase()
263 GetDevice(test_case->num_ranks_per_node, node, local_rank); in VerifyResults()
333 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in RunMultiNodeAllReduceTest()
382 auto* device = GetDevice(num_ranks_per_node, node, local_rank); in RunMultiNodeBroadcastTest()
425 static BaseGPUDevice* GetDevice(int num_ranks_per_node, int node, in GetDevice() function in tensorflow::NcclManagerTest
476 auto* device = this->GetDevice(num_ranks, /*node=*/0, rank); in TYPED_TEST()
523 auto* device = this->GetDevice(num_ranks, /*node=*/0, rank); in TYPED_TEST()
543 auto* device = this->GetDevice(num_ranks, /*node=*/0, rank); in TYPED_TEST()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_formation.cc58 StringRef GetDevice(Operation* op) { in GetDevice() function
94 GetDevice(defining_op) == c.device) in CanMergeIntoCluster()
190 auto device = GetDevice(&op); in BuildClusters()
/external/pdfium/core/fpdfapi/render/
Dcpdf_scaledrenderbuffer.h33 CFX_RenderDevice* GetDevice() const;
Dcpdf_scaledrenderbuffer.cpp68 CFX_RenderDevice* CPDF_ScaledRenderBuffer::GetDevice() const { in GetDevice() function in CPDF_ScaledRenderBuffer
Dcpdf_renderstatus.cpp364 CPDF_RenderStatus status(m_pContext.Get(), buffer.GetDevice()); in DrawObjWithBackground()
/external/mesa3d/include/D3D9/
Dd3d9.h117 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
314 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
320 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
329 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
351 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
380 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
386 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
392 virtual HRESULT WINAPI GetDevice(IDirect3DDevice9 **ppDevice) = 0;
602 HRESULT (WINAPI *GetDevice)(IDirect3DBaseTexture9 *This, IDirect3DDevice9 **ppDevice);
628 #define IDirect3DBaseTexture9_GetDevice(p,a) (p)->lpVtbl->GetDevice(p,a)
[all …]
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_virtual_mem_allocator.cc71 TF_RETURN_IF_ERROR(GpuDriver::GetDevice(gpu_id.value(), &gpu_handle)); in Create()
77 TF_RETURN_IF_ERROR(GpuDriver::GetDevice(peer_id.value(), &peer_handle)); in Create()
/external/deqp-deps/amber/src/vulkan/
Dpipeline.h68 Device* GetDevice() const { return device_; } in GetDevice() function
Dgraphics_pipeline.cc364 pipeline_->GetDevice()->GetPtrs()->vkCmdBeginRenderPass( in RenderPassGuard()
372 pipeline_->GetDevice()->GetPtrs()->vkCmdEndRenderPass( in ~RenderPassGuard()
/external/angle/src/libANGLE/renderer/d3d/d3d11/
DStreamProducerD3DTexture.cpp123 textureD3D->GetDevice(&device); in validateD3DTexture()
/external/angle/src/libANGLE/renderer/gl/wgl/
DD3DTextureSurfaceWGL.cpp59 texture11->GetDevice(&d3d11Device); in GetD3D11TextureInfo()
151 HRESULT result = texture9->GetDevice(&d3d9Device); in GetD3D9TextureInfo()
/external/angle/src/tests/egl_tests/
DEGLPresentPathD3D11Test.cpp245 mOffscreenSurfaceD3D11Texture->GetDevice(&device); in checkPixelsUsingD3D()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector.cc364 FindDeviceIndex(node_view, fanout.node_view()->GetDevice(), &index); in SelectDeviceIndex()
Dgeneric_layout_optimizer_transposer_test.cc529 EXPECT_EQ(const_perm_node->GetDevice(), kDevice); in TEST_F()
584 EXPECT_EQ(transpose_node->GetDevice(), kDevice); in TEST_F()
Dgeneric_layout_optimizer_transposer.cc1289 const string& node_device = node->GetDevice(); in MaybeReshapeVectorFanin()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_driver.h206 static port::Status GetDevice(int device_ordinal, GpuDeviceHandle* device);
Dasm_compiler.cc145 TF_RETURN_IF_ERROR(gpu::GpuDriver::GetDevice(device_ordinal, &handle)); in CompileGpuAsm()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view_internal.h165 const string& GetDevice() const { return node()->device(); } in GetDevice() function
434 if (diff->graph_view->GetNode(diff->node_index)->GetDevice() == device) { in UpdateDevice()
Dgraph_view_test.cc287 TYPED_TEST(TypedNodeViewTest, GetDevice) { in TYPED_TEST() argument
299 EXPECT_EQ(a_node->GetDevice(), "device_a"); in TYPED_TEST()
303 EXPECT_EQ(b_node->GetDevice(), "device_b"); in TYPED_TEST()
307 EXPECT_EQ(c_node->GetDevice(), "device_c"); in TYPED_TEST()
311 EXPECT_EQ(d_node->GetDevice(), ""); in TYPED_TEST()
812 EXPECT_EQ(node_view->GetDevice(), expected_node_view.GetDevice()); in CompareGraphViewWithGraph()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc169 status = GpuDriver::GetDevice(device_ordinal_, &device_); in Init()
812 auto status = GpuDriver::GetDevice(device_ordinal, &device); in CreateDeviceDescription()
Drocm_driver.cc321 /* static */ port::Status GpuDriver::GetDevice(int device_ordinal, in GetDevice() function in stream_executor::gpu::GpuDriver
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc139 status = GpuDriver::GetDevice(device_ordinal_, &device_); in Init()
964 auto status = GpuDriver::GetDevice(device_ordinal, &device); in CreateDeviceDescription()
Dcuda_driver.cc347 /* static */ port::Status GpuDriver::GetDevice(int device_ordinal, in GetDevice() function in stream_executor::gpu::GpuDriver
/external/webrtc/modules/audio_device/win/
Dcore_audio_utility_win.cc316 error = device_enum->GetDevice(rtc::ToUtf16(device_id).c_str(), in CreateDeviceInternal()
/external/angle/src/libANGLE/renderer/d3d/d3d9/
DRenderer9.cpp747 texture->GetDevice(&textureDevice); in getD3DTextureInfo()

12