mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8f5cdc6d5d
Summary: Re-land https://github.com/pytorch/pytorch/pull/76711 by fixing internal build errors. Generate class-level opkind as a static method instead of a static member. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77102 Approved by: https://github.com/wconstab, https://github.com/JackCaoG, https://github.com/antoniojkim
36 lines
840 B
C++
36 lines
840 B
C++
#pragma once
|
|
|
|
#include <c10/core/Scalar.h>
|
|
#include <torch/csrc/lazy/ts_backend/ts_node.h>
|
|
|
|
namespace torch {
|
|
namespace lazy {
|
|
|
|
// Differently from Constant, this is a scalar value broadcasted to a shape.
|
|
// Even though a Constant could have been used, for simple scalars broadcasted
|
|
// to big shapes, the Constant leads to big literals expanded within the
|
|
// computation graph.
|
|
class TORCH_API Scalar : public TsNode {
|
|
public:
|
|
static OpKind ClassOpKind() {
|
|
return OpKind(at::prim::Constant);
|
|
}
|
|
|
|
Scalar(const at::Scalar& value, Shape shape);
|
|
Scalar(const at::Scalar& value, c10::ScalarType type);
|
|
|
|
std::string ToString() const override;
|
|
|
|
const at::Scalar& value() const {
|
|
return value_;
|
|
}
|
|
|
|
private:
|
|
at::Scalar value_;
|
|
};
|
|
|
|
TORCH_API hash_t ScalarHash(const at::Scalar& s);
|
|
|
|
} // namespace lazy
|
|
} // namespace torch
|