Searched refs:computeShape (Results 1 – 3 of 3) sorted by relevance
123 shapes_.push_back(computeShape(shape_fn)); in addComputedShape()129 Shape Node::computeShape(const std::function<Shape()>& shape_fn) { in computeShape() function in torch::lazy::Node
131 Shape computeShape(const std::function<Shape()>& shape_fn);
89 EXPECT_EQ(node.shape(), node.computeShape([&]() { return node.shape(); })); in TEST()