Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A 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()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ops/
H A Dgeneric.cpp18 const std::function<Shape()>& shape_fn, in Generic()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_node.cpp57 const std::function<Shape()>& shape_fn, in TsNode()