Searched refs:input_device_context (Results 1 – 5 of 5) sorted by relevance
88 {ctx->input_device_context(0), ctx->input_device_context(0)}) {
103 col_ctx_->op_ctx->input_device_context(0), in Run()
91 col_ctx_->op_ctx->input_device_context(0), in Run()
1021 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() function1627 inline DeviceContext* OpKernelContext::input_device_context(int index) { in input_device_context() function
538 static_cast<GPUDeviceContext*>(context->input_device_context(i)); in ComputeHelper()