Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_map.cpp296 for (const auto& shape_symbol : shape_symbols.value()) { in PrintMaps() local
297 if (shape_symbol.is_static()) { in PrintMaps()
298 ss << shape_symbol.static_size() << ", "; in PrintMaps()
H A Dshape_type_inference.cpp799 auto shape_symbol = input_shape_value.value()[idx]; in ProcessShapeForConcatNode() local
800 if (shape_symbol.is_static()) { in ProcessShapeForConcatNode()
801 size_total += shape_symbol.static_size(); in ProcessShapeForConcatNode()
821 auto shape_symbol = input_shape_value.value()[idx]; in ProcessShapeForConcatNode() local
822 if (shape_symbol.is_static()) { in ProcessShapeForConcatNode()
824 shape_symbol.static_size())); in ProcessShapeForConcatNode()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dschema_type_parser.cpp309 bool shape_symbol = false; in parseRefinedTensor() local
314 shape_symbol = true; in parseRefinedTensor()
325 if (shape_symbol) { in parseRefinedTensor()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dsymbolic_shape_analysis.cpp455 auto shape_symbol = *maybe_shape_symbol; in substituteSymbolicProperties() local
458 use.user->output(), shape_symbol.value()); in substituteSymbolicProperties()
460 int64_t symbolic_index = shape_symbol.value(); in substituteSymbolicProperties()