| 80 | } |
| 81 | |
| 82 | SubContext::SubContext(OpKernelContext* ctx, OpKernelContext::Params* params, |
| 83 | OpKernel* op, Tensor* output, Tensor* input) |
| 84 | : sub_params_(*params), |
| 85 | sub_inputs_({TensorValue(output), TensorValue(input)}), |
| 86 | sub_input_attr_({ctx->input_alloc_attr(0), ctx->input_alloc_attr(0)}) { |
| 87 | sub_params_.op_kernel = op; |
| 88 | sub_params_.inputs = &sub_inputs_; |
| 89 | sub_params_.input_alloc_attrs = &sub_input_attr_; |
| 90 | sub_params_.op_device_context = ctx->op_device_context(); |
| 91 | sub_params_.eigen_gpu_device = nullptr; |
| 92 | sub_params_.ensure_eigen_gpu_device(); |
| 93 | sub_params_.forward_from_array = &forward_from_; |
| 94 | sub_ctx_.reset(new OpKernelContext(&sub_params_, 1)); |
| 95 | } |
| 96 | |
| 97 | Status ComputeBinOp(OpKernelContext* op_ctx, OpKernelContext::Params* params, |
| 98 | Device* device, OpKernel* op, Tensor* output, |
nothing calls this directly
no test coverage detected