Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dconstant_map.h48 static std::optional<std::vector<int64_t>> GetShapeInto1DInt64Vector( in C10_DIAGNOSTIC_POP()
H A Dconstant_map.cpp109 std::optional<std::vector<int64_t>> ConstantValueMap::GetShapeInto1DInt64Vector( in GetShapeInto1DInt64Vector() function in torch::jit::ConstantValueMap
H A Dshape_type_inference.cpp997 auto output_rank = ConstantValueMap::GetShapeInto1DInt64Vector(shape_name); in ProcessReshapeNode()
1289 ConstantValueMap::GetShapeInto1DInt64Vector(n->input()->debugName()); in ComputeConstant()