Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/functorch/dim/
H A Dreference.py534 tensor_insert_point = None
543 if tensor_insert_point is None:
544 tensor_insert_point = len(result_levels)
555 if tensor_insert_point is not None:
556 result_levels[tensor_insert_point:tensor_insert_point] = index_levels
/aosp_15_r20/external/pytorch/functorch/csrc/dim/
H A Ddim.cpp2521 int64_t tensor_insert_point = -1; in getsetitem_flat() local
2524 if (tensor_insert_point == -1) { in getsetitem_flat()
2525 tensor_insert_point = result_levels.size(); in getsetitem_flat()
2526 } else if (tensor_insert_point != result_levels.size()) { in getsetitem_flat()
2527 tensor_insert_point = 0; in getsetitem_flat()
2569 if (tensor_insert_point != -1) { in getsetitem_flat()
2570 …result_levels.insert(A, result_levels.slice(tensor_insert_point, tensor_insert_point), index_level… in getsetitem_flat()