OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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