Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/mobile/nnc/
Dcontext.h95 struct TORCH_API SymbolicShapePosition { struct
96 SymbolicShapePosition() = default;
97 SymbolicShapePosition(int64_t input_idx, int64_t dim_idx) in SymbolicShapePosition() argument
171 const std::vector<SymbolicShapePosition>& sym_shape_positions() const { in sym_shape_positions()
176 const std::vector<SymbolicShapePosition>& sym_shape_pos) { in set_sym_shape_positions()
188 std::vector<SymbolicShapePosition> sym_shape_positions_;