Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp924 std::vector<int64_t> axes_vector(rank_0); in ProcessReduceNode() local
1031 const std::vector<int64_t>& axes_vector, in ComputeShapeForSlice()
1104 std::vector<int64_t> axes_vector(input0_shape_value.size(), 0); in ProcessSliceNode() local