Searched refs:default_context (Results 1 – 17 of 17) sorted by relevance
321 default_context = context.context()324 for callback in default_context.post_execution_callbacks:339 default_context.clear_post_execution_callbacks()341 default_context.add_post_execution_callback(callback)343 default_context.add_post_execution_callback(
107 src_dev_ctx = dev_info->default_context; in MemCpyAsync()113 dst_dev_ctx = dev_info->default_context; in MemCpyAsync()
439 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()473 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()618 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()644 dev_ctx = dev_info->default_context; in DoBroadcast()
310 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()430 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()456 dev_ctx = dev_info->default_context; in DoGather()
330 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()454 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()485 dev_ctx = dev_info->default_context; in DoReduce()
143 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
179 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
110 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()134 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
258 gpu_device_info->default_context->CopyDeviceTensorToCPU( in TensorDebugString()
172 *device_context = dev_info->default_context; in GetDeviceContext()
183 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()239 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()263 dev_ctx = dev_info->default_context; in DoReduce()
286 src_device_context = srcd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()290 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
226 device_context = device_->tensorflow_gpu_device_info()->default_context; in Run()
153 DeviceContext* default_context = nullptr; member
1036 if (dev_info) ret = dev_info->default_context; in op_device_context()
341 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()
396 gpu_device_info_->default_context = device_contexts_[0]; in Init()