HomeSort by relevance Sort by last modified time
    Searched refs:default_context (Results 1 - 17 of 17) sorted by null

  /external/tensorflow/tensorflow/python/eager/
execution_callbacks.py 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(
  /external/tensorflow/tensorflow/core/common_runtime/
collective_rma_local.cc 107 src_dev_ctx = dev_info->default_context;
113 dst_dev_ctx = dev_info->default_context;
hierarchical_tree_broadcaster_test.cc 439 dev_info->default_context->CopyDeviceTensorToCPU(
473 dev_info->default_context->CopyDeviceTensorToCPU(
618 dev_info->default_context->CopyCPUTensorToDevice(
644 dev_ctx = dev_info->default_context;
    [all...]
ring_gatherer.cc 143 Status s = gpu_info->default_context->ThenExecute(
ring_reducer.cc 179 Status s = gpu_info->default_context->ThenExecute(
ring_gatherer_test.cc 310 dev_info->default_context->CopyDeviceTensorToCPU(
430 dev_info->default_context->CopyCPUTensorToDevice(
456 dev_ctx = dev_info->default_context;
ring_reducer_test.cc 330 dev_info->default_context->CopyDeviceTensorToCPU(
454 dev_info->default_context->CopyCPUTensorToDevice(
485 dev_ctx = dev_info->default_context;
ring_alg.cc 258 gpu_device_info->default_context->CopyDeviceTensorToCPU(
process_function_library_runtime_test.cc 110 gpu_device_->tensorflow_gpu_device_info()->default_context;
134 gpu_device_->tensorflow_gpu_device_info()->default_context;
process_function_library_runtime.cc 172 *device_context = dev_info->default_context;
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
collective_nccl_reducer_test.cc 183 dev_info->default_context->CopyDeviceTensorToCPU(
239 dev_info->default_context->CopyCPUTensorToDevice(
263 dev_ctx = dev_info->default_context;
  /external/tensorflow/tensorflow/core/framework/
device_base.h 143 // "default_context" is used by OpKernelContext whenever a device does not
153 DeviceContext* default_context = nullptr; member in struct:tensorflow::DeviceBase::GpuDeviceInfo
op_kernel.h     [all...]
  /external/tensorflow/tensorflow/core/common_runtime/eager/
tensor_handle.cc 286 src_device_context = srcd->tensorflow_gpu_device_info()->default_context;
290 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context;
kernel_and_device.cc 226 device_context = device_->tensorflow_gpu_device_info()->default_context;
  /external/tensorflow/tensorflow/compiler/jit/
xla_device.cc 341 gpu_device_info->default_context = device_context_;
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 396 gpu_device_info_->default_context = device_contexts_[0];
    [all...]

Completed in 1225 milliseconds