Searched defs:matrix_dims (Results 1 – 3 of 3) sorted by relevance
51 .ReadResult <- function (f, entry_size, matrix_dims) { argument99 matrix_dims <- dim(joint_conditional[[1]]) functionVar
69 auto matrix_dims = a_shape.dimensions().subspan( in CholeskyUnblocked() local
3202 ArrayRef<int64_t> matrix_dims = type.getShape().take_back(2); in BroadcastBatchMatMulV2Operands() local