Searched refs:matrix_element (Results 1 – 1 of 1) sorted by relevance
169 auto matrix_element = GetTupleElement(tuple_data, 1); in XLA_TEST_F() local171 auto matrix_shape = builder.GetShape(matrix_element).ConsumeValueOrDie(); in XLA_TEST_F()172 Add(matrix_element, vector_element, in XLA_TEST_F()