Searched refs:contracted_shape (Results 1 – 1 of 1) sorted by relevance
41 std::vector<int64> contracted_shape(input_shape.dimensions().begin(), in Contract() local43 contracted_shape[dim] *= 4; in Contract()45 return xla::Reshape(xla::Transpose(input, permutation), contracted_shape); in Contract()