Home
last modified time | relevance | path

Searched refs:input_device_context (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc88 {ctx->input_device_context(0), ctx->input_device_context(0)}) {
Dring_gatherer.cc103 col_ctx_->op_ctx->input_device_context(0), in Run()
Dring_reducer.cc91 col_ctx_->op_ctx->input_device_context(0), in Run()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1021 T* input_device_context(int index);
1022 DeviceContext* input_device_context(int index);
1618 T* OpKernelContext::input_device_context(int index) { in input_device_context() function
1627 inline DeviceContext* OpKernelContext::input_device_context(int index) { in input_device_context() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc538 static_cast<GPUDeviceContext*>(context->input_device_context(i)); in ComputeHelper()