Method Compile

tensorflow/compiler/tf2xla/kernels/xla_dot_op.cc:44–53

  }

  void Compile(XlaOpKernelContext* context) override {
    const TensorShape lhs_shape = context->InputShape(0);
    const TensorShape rhs_shape = context->InputShape(1);

    // We do only minimal checking, relying on XLA to check the shape
    // invariants.
    xla::XlaOp output = xla::DotGeneral(context->Input(0), context->Input(1),
                                        dnums_, &precision_config_);
    context->SetOutput(0, output);
  }

 private:
  xla::DotDimensionNumbers dnums_;