Method Compile

tensorflow/compiler/tf2xla/kernels/empty_op.cc:39–55

  }

  void Compile(XlaOpKernelContext* ctx) override {
    // The output of this Op is a tensor of shape 'shape' with each
    // element set to the default value of 'dtype'. If 'init' is false then
    // the result values may be left undefined, though we don't do that here.
    const TensorShape shape_shape = ctx->InputShape("shape");
    OP_REQUIRES(
        ctx, TensorShapeUtils::IsVector(shape_shape),
        errors::InvalidArgument("shape must be a vector of int32, got shape ",
                                shape_shape.DebugString()));

    std::vector<int64> shape;
    OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector("shape", &shape));

    auto default_value = xla::Zero(ctx->builder(), type_);
    auto result = xla::Broadcast(default_value, shape);
    ctx->SetOutput(0, result);
  }

 private:
  DataType dtype_;