Searched refs:reshaped_result (Results 1 – 2 of 2) sorted by relevance
168 XlaOp reshaped_result = Collapse(transposed_result, {0, 1});172 return reshaped_result;180 return Transpose(reshaped_result, result_dimensions);
403 Value reshaped_result = if_builder.create<tensor::CastOp>( in createRankSpecializedBroadcastAndOp() local405 if_builder.create<scf::YieldOp>(loc, reshaped_result); in createRankSpecializedBroadcastAndOp()