Searched refs:tensor_layout1 (Results 1 – 1 of 1) sorted by relevance
283 def _check_similar_layout(tensor_layout1, tensor_layout2): argument285 if tensor_layout1[1] != tensor_layout2[1]:287 for i in tensor_layout1[1]:290 if tensor_layout1[0][-1-i] != tensor_layout2[0][-1-i]:295 def _check_same_layout(tensor_layout1, tensor_layout2): argument297 return tensor_layout1[0] == tensor_layout2[0] and tensor_layout1[1] == tensor_layout2[1]