Searched refs:a2_transpose_shape (Results 1 – 1 of 1) sorted by relevance
1157 SmallVector<int64_t, 2> a2_transpose_shape(block_rank + input_rank); in convertBatchToSpaceNDOp() local1171 a2_transpose_shape[i] = a1_shape[a2_perm[i]]; in convertBatchToSpaceNDOp()1178 RankedTensorType::get(ArrayRef<int64_t>(a2_transpose_shape), in convertBatchToSpaceNDOp()