Searched refs:transposed_rhs_shape (Results 1 – 2 of 2) sorted by relevance
251 TensorShape transposed_rhs_shape(rhs.shape()); in ComputeAsync() local252 transposed_rhs_shape.RemoveLastDims(2); in ComputeAsync()253 transposed_rhs_shape.AddDim(nrhs); in ComputeAsync()254 transposed_rhs_shape.AddDim(n); in ComputeAsync()259 transposed_rhs_shape, &transposed_rhs), in ComputeAsync()
4395 Shape transposed_rhs_shape = ShapeUtil::MakeShape(element_type, {n, k}); in TEST_P() local4405 1, transpose_rhs ? transposed_rhs_shape : rhs_shape, "rhs")); in TEST_P()