Searched refs:reshaped_x (Results 1 – 1 of 1) sorted by relevance
467 XlaOp reshaped_x = xla::Reshape(x_result, reshaped_dims, reshaped_sizes); in CalculateGradWarp() local479 return xla::ConcatInDim(ctx->builder(), {reshaped_x, reshaped_y}, in CalculateGradWarp()