Searched refs:zero_matrix (Results 1 – 3 of 3) sorted by relevance
79 zero_matrix = array_ops.zeros(shape, dtype=dtype)80 return array_ops.matrix_set_diag(zero_matrix, diag_ones)
409 // zero_matrix = { [I, J, K,...M, N] zero matrix }410 // return (indicator ? input : zero_matrix)
1405 Value zero_matrix = rewriter.create<BroadcastOp>( in matchAndRewrite() local1410 compare, reshaped_input, zero_matrix); in matchAndRewrite()