Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dexecution_callbacks.py321 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(
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local.cc107 src_dev_ctx = dev_info->default_context; in MemCpyAsync()
113 dst_dev_ctx = dev_info->default_context; in MemCpyAsync()
Dhierarchical_tree_broadcaster_test.cc439 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()
Dring_gatherer_test.cc310 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
430 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
456 dev_ctx = dev_info->default_context; in DoGather()
Dring_reducer_test.cc330 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
454 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
485 dev_ctx = dev_info->default_context; in DoReduce()
Dring_gatherer.cc143 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
Dring_reducer.cc179 Status s = gpu_info->default_context->ThenExecute( in RunAsyncParts()
Dprocess_function_library_runtime_test.cc110 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()
134 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
Dring_alg.cc258 gpu_device_info->default_context->CopyDeviceTensorToCPU( in TensorDebugString()
Dprocess_function_library_runtime.cc172 *device_context = dev_info->default_context; in GetDeviceContext()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc183 dev_info->default_context->CopyDeviceTensorToCPU( in RunTest()
239 dev_info->default_context->CopyCPUTensorToDevice( in InitTensor()
263 dev_ctx = dev_info->default_context; in DoReduce()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc286 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()
Dkernel_and_device.cc226 device_context = device_->tensorflow_gpu_device_info()->default_context; in Run()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h153 DeviceContext* default_context = nullptr; member
Dop_kernel.h1036 if (dev_info) ret = dev_info->default_context; in op_device_context()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc341 gpu_device_info->default_context = device_context_; in GetDeviceContextLocked()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc396 gpu_device_info_->default_context = device_contexts_[0]; in Init()