Home
last modified time | relevance | path

Searched refs:computeShape (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/lazy/core/
Dir.cpp123 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
Dir.h131 Shape computeShape(const std::function<Shape()>& shape_fn);
/external/pytorch/test/cpp/lazy/
Dtest_cache.cpp89 EXPECT_EQ(node.shape(), node.computeShape([&]() { return node.shape(); })); in TEST()