Searched refs:a2_transpose_shape (Results 1 – 1 of 1) sorted by relevance
1098 SmallVector<int64_t> a2_transpose_shape(block_rank + input_rank); in convertBatchToSpaceNDOp() local1112 a2_transpose_shape[i] = a1_shape[a2_perm[i]]; in convertBatchToSpaceNDOp()1122 RankedTensorType::get(a2_transpose_shape, result_type.getElementType()), in convertBatchToSpaceNDOp()