Searched defs:entry_shape (Results 1 – 2 of 2) sorted by relevance
1101 at::IntArrayRef entry_shape = tensors[0].sizes(); in compute_shape_stack() local
2672 at::IntArrayRef entry_shape = tensors[0].sizes(); in get_stack_inputs() local2718 at::IntArrayRef entry_shape = tensors[0].sizes(); in check_stack_inputs() local