Searched refs:a4_size_vals (Results 1 – 1 of 1) sorted by relevance
1212 SmallVector<int64_t, 4> a4_begin_vals(input_rank), a4_size_vals(input_rank); in convertBatchToSpaceNDOp() local1218 a4_size_vals[i] = result_type.getShape()[i]; in convertBatchToSpaceNDOp()1223 a4_size_vals[i] = a4_shape[i] - crops[i - 1].first - crops[i - 1].second; in convertBatchToSpaceNDOp()1230 RankedTensorType::get(ArrayRef<int64_t>(a4_size_vals), in convertBatchToSpaceNDOp()1233 rewriter.getI64ArrayAttr(a4_size_vals)) in convertBatchToSpaceNDOp()