Home
last modified time | relevance | path

Searched defs:shape_fn (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/lazy/core/
Dir.cpp95 const std::function<Shape()>& shape_fn, in Node()
122 void Node::addComputedShape(const std::function<Shape()>& shape_fn) { in addComputedShape()
129 Shape Node::computeShape(const std::function<Shape()>& shape_fn) { in computeShape()
/external/pytorch/torch/csrc/lazy/ts_backend/ops/
Dgeneric.cpp18 const std::function<Shape()>& shape_fn, in Generic()
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_node.cpp57 const std::function<Shape()>& shape_fn, in TsNode()