Searched refs:s_matrix (Results 1 – 2 of 2) sorted by relevance
1100 auto s_matrix = create(&t); in TEST_F() local1102 s_matrix, "Input tensor must be rank 1, but was rank 2")) in TEST_F()1103 << s_matrix; in TEST_F()
3722 s_matrix = array_ops.stack([3730 array_ops.expand_dims(w_g, axis=1) * s_matrix, axis=2)