Method ValidateInputsAreSameShape

tensorflow/compiler/tf2xla/xla_op_kernel.cc:36–38

    : context_(context), dynamic_dimension_is_minus_one_(false) {}

bool XlaOpKernelContext::ValidateInputsAreSameShape(OpKernel* op) {
  return context_->ValidateInputsAreSameShape(op);
}

XlaContext* XlaOpKernelContext::xla_context() const {
  return &XlaContext::Get(context_);