Searched refs:GetDeviceId (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_kernel.cc | 119 RETURN_IF_ERROR(GetKernelPrivateMemorySize(kernel_, program.GetDeviceId(), in CreateFromProgram()
121 RETURN_IF_ERROR(GetKernelMaxWorkGroupSize(kernel_, program.GetDeviceId(), in CreateFromProgram()
|
| D | cl_program.h | 56 cl_device_id GetDeviceId() const { return device_id_; } in GetDeviceId() function
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_interface.h | 63 virtual CUptiResult GetDeviceId(CUcontext context, uint32* deviceId) = 0;
|
| D | cupti_wrapper.h | 58 CUptiResult GetDeviceId(CUcontext context, uint32* deviceId) override;
|
| D | cupti_wrapper.cc | 59 CUptiResult CuptiWrapper::GetDeviceId(CUcontext context, uint32* deviceId) { in GetDeviceId() function in tensorflow::profiler::CuptiWrapper
|
| D | cupti_tracer.cc | 447 cupti_interface->GetDeviceId(p2p_params->srcContext, &src_device); in AddP2PMemcpyEventUponApiExit()
448 cupti_interface->GetDeviceId(p2p_params->dstContext, &dst_device); in AddP2PMemcpyEventUponApiExit()
1806 cupti_interface_->GetDeviceId(cbdata->context, &device_id)); in HandleCallback()
1938 RETURN_IF_CUPTI_ERROR(cupti_interface_->GetDeviceId(context, &device_id)); in ProcessActivityBuffer()
|