Searched refs:get_dim_size (Results 1 – 2 of 2) sorted by relevance
219 print("dim size:", vector.get_dim_size(1))304 invalid_get_dim_size = unranked_tensor.get_dim_size(1)372 invalid_get_dim_size = unranked_memref.get_dim_size(1)
868 def get_dim_size(dim): function874 num_rows = get_dim_size(-2)875 num_cols = get_dim_size(-1)