Home
last modified time | relevance | path

Searched refs:default_context (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc142 src_dev_ctx = dev_info->default_context; in MemCpyAsync()
148 dst_dev_ctx = dev_info->default_context; in MemCpyAsync()
Dpermuter_test.cc289 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
364 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
386 dev_ctx = dev_info->default_context; in DoPermute()
Dhierarchical_tree_broadcaster_test.cc453 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
482 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
622 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
644 dev_ctx = dev_info->default_context; in DoBroadcast()
Dring_gatherer_test.cc321 CHECK(dev_info->default_context in RunTest()
440 CHECK(dev_info->default_context in InitTensor()
464 dev_ctx = dev_info->default_context; in DoGather()
Dring_reducer_test.cc343 CHECK(dev_info->default_context in RunTest()
466 CHECK(dev_info->default_context in InitTensor()
496 dev_ctx = dev_info->default_context; in DoReduce()
Dring_gatherer.cc150 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
Dring_reducer.cc200 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
Dring_alg.cc259 Status st = gpu_device_info->default_context->CopyDeviceTensorToCPUSync( in TensorDebugString()
Dprocess_function_library_runtime_test.cc179 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()
197 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
Dprocess_function_library_runtime.cc202 *device_context = dev_info->default_context; in GetDeviceContext()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc89 device_->tensorflow_gpu_device_info()->default_context; in RunComposedOp()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_test.cc217 TF_CHECK_OK(dev_info->default_context->CopyDeviceTensorToCPUSync( in RunTest()
280 TF_CHECK_OK(dev_info->default_context->CopyCPUTensorToDeviceSync( in InitTensor()
290 dev_ctx = dev_info->default_context; in PrepareDeviceContext()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h76 context(static_cast<GPUDeviceContext*>(info->default_context)), in Participant()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h156 DeviceContext* default_context = nullptr; member
Dop_kernel.h1071 if (dev_info) ret = dev_info->default_context; in op_device_context()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc367 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()
Dxla_launch_util.cc599 if (dev_info) device_context = dev_info->default_context; in BuildXlaCompilerArguments()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc1006 src_device_context = srcd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
1010 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc415 DeviceContext* device_context = device_info->default_context; in TEST_F()
Dgpu_device.cc474 gpu_device_info_->default_context = device_context_; in Init()
/external/mesa3d/include/CL/
Dcl2.hpp2992 static Context setDefault(const Context &default_context) in setDefault() argument
2994 std::call_once(default_initialized_, makeDefaultProvided, std::cref(default_context)); in setDefault()