Searched defs:sliced_tensor (Results 1 – 2 of 2) sorted by relevance
84 auto sliced_tensor = at_tensor.slice(1, 0, 2); in TEST() local109 auto sliced_tensor = at_tensor.slice(1, 0, 2); in TEST() local
249 auto sliced_tensor = wei.chunk(4, 0); in permute_wei_for_miopen() local253 auto sliced_tensor = wei.chunk(3, 0); in permute_wei_for_miopen() local