Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp1290 std::vector<ExprHandle> new_axes( in convertSymbolicOutputToCorrectStrides() local
1403 std::vector<ExprHandle> new_axes(sorted_stride_indices.size()); in convertStaticShapeOutputToCorrectStrides() local
H A Dlowerings.cpp1801 std::vector<VarHandle> new_axes; in nnc_lowerings_lazy_registration() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/
H A Dmisc.cpp409 std::vector<VarHandle> new_axes; in computeReshape() local