Home
last modified time | relevance | path

Searched refs:sizesFromSymbolicShape (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/tensorexpr/
Dkernel.cpp491 std::vector<ExprHandle> TensorExprKernel::sizesFromSymbolicShape( in sizesFromSymbolicShape() function in TensorExprKernel
513 return sizesFromSymbolicShape(tt->symbolic_sizes()); in sizesForValue()
1148 auto size_handles = sizesFromSymbolicShape(tt->symbolic_sizes()); in bindInput()
1159 sizesFromSymbolicShape(tt->symbolic_sizes()), in bindInput()
1330 auto sizes = sizesFromSymbolicShape(tt->symbolic_sizes()); in convertSymbolicOutputToCorrectStrides()
1788 auto sizes = sizesFromSymbolicShape(tt->symbolic_sizes()); in compile()
Dkernel.h291 std::vector<ExprHandle> sizesFromSymbolicShape(