Searched defs:new_axis_pos (Results 1 – 2 of 2) sorted by relevance
73 … std::vector<int64_t> *new_axis_pos, std::vector<int64_t> *shrink_axis_pos) { in GetAndCheckAttrMaskV2()184 std::vector<int64_t> new_axis_pos; in EllipsisInferShapeV2() local241 std::vector<int64_t> new_axis_pos; in ComputeInferShapeV2() local309 std::vector<int64_t> new_axis_pos; in DynamicComputeInferShapeV2() local
70 std::vector<int64_t> new_axis_pos; member