mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a2396b2dd8
Follows #137407 Pull Request resolved: https://github.com/pytorch/pytorch/pull/137459 Approved by: https://github.com/Skylion007
35 lines
913 B
C++
35 lines
913 B
C++
#include <torch/csrc/lazy/ts_backend/ops/generic.h>
|
|
|
|
namespace torch::lazy {
|
|
|
|
Generic::Generic(
|
|
OpKind op,
|
|
OpList operands,
|
|
Shape shape,
|
|
size_t num_outputs,
|
|
hash_t hash_seed)
|
|
: TsNode(op, operands, {std::move(shape)}, num_outputs, hash_seed),
|
|
hash_seed_(hash_seed) {}
|
|
|
|
Generic::Generic(
|
|
OpKind op,
|
|
OpList operands,
|
|
const std::function<Shape()>& shape_fn,
|
|
size_t num_outputs,
|
|
hash_t hash_seed)
|
|
: TsNode(op, operands, shape_fn, num_outputs, hash_seed),
|
|
hash_seed_(hash_seed) {}
|
|
|
|
Generic::Generic(
|
|
OpKind op,
|
|
OpList operands,
|
|
size_t num_outputs,
|
|
hash_t hash_seed)
|
|
: TsNode(op, operands, num_outputs, hash_seed), hash_seed_(hash_seed) {}
|
|
|
|
Generic::Generic(OpKind op, Shape shape, size_t num_outputs, hash_t hash_seed)
|
|
: TsNode(op, std::move(shape), num_outputs, hash_seed),
|
|
hash_seed_(hash_seed) {}
|
|
|
|
} // namespace torch::lazy
|