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