File tree Expand file tree Collapse file tree 2 files changed +0
-28
lines changed Expand file tree Collapse file tree 2 files changed +0
-28
lines changed Original file line number Diff line number Diff line change @@ -84,16 +84,6 @@ torch::lazy::hash_t GetOperandHashes(const OpList& operands,
8484
8585} // namespace
8686
87- // const xla::Shape& XlaValue::xla_shape() const {
88- // XlaNode* casted = dynamic_cast<XlaNode*>(node.get());
89- // return casted->xla_shape(index);
90- // }
91-
92- // const xla::Shape& XlaValue::xla_node_shape() const {
93- // XlaNode* casted = dynamic_cast<XlaNode*>(node.get());
94- // return casted->xla_shape();
95- // }
96-
9787XlaNode::XlaNode (torch::lazy::OpKind op, OpList operands,
9888 std::vector<torch::lazy::Shape>&& shapes, xla::Shape xla_shape,
9989 size_t num_outputs, torch::lazy::hash_t hash_seed)
Original file line number Diff line number Diff line change @@ -34,24 +34,6 @@ template <typename T>
3434using OutputMap =
3535 std::unordered_map<torch::lazy::Output, T, torch::lazy::Output::Hasher>;
3636
37- // Represents an input/operand for a XlaNode object.
38- // struct XlaValue : public torch::lazy::Value {
39- // torch::lazy::Value() = default;
40- // torch::lazy::Value(torch::lazy::NodePtr node, size_t index = 0)
41- // :
42- // torch::lazy::Value(std::dynamic_pointer_cast<torch::lazy::Node>(node),
43- // index) {}
44-
45- // // Retrieves the shape of this value. If the IR XlaNode generating the
46- // value
47- // // is a multi-output node, the shape returned by this API will not be the
48- // full
49- // // tuple shape, but only the shape at index referred by this value.
50- // // To retrieve the full tuple shape in that case, use the node_shape() API.
51- // // const xla::Shape& xla_shape() const;
52- // // const xla::Shape& xla_node_shape() const;
53- // };
54-
5537using OpList = absl::Span<const torch::lazy::Value>;
5638
5739// A node in the graph. Nodes for operations which requires extra data to be
You can’t perform that action at this time.
0 commit comments