Searched refs:add_bitcast (Results 1 – 1 of 1) sorted by relevance
3594 auto add_bitcast = [&](Shape shape, HloInstruction* operand) { in SimplifyConvToDot() local3627 auto new_lhs = add_bitcast(new_input_shape, lhs); in SimplifyConvToDot()3628 auto new_rhs = add_bitcast(new_filter_shape, rhs); in SimplifyConvToDot()3637 ReplaceInstruction(convolution, add_bitcast(convolution_shape, dot))); in SimplifyConvToDot()