Searched refs:new_input_shape (Results 1 – 3 of 3) sorted by relevance
145 Shape* new_input_shape; in PadForTensorCores() local148 std::tie(new_input_shape, new_filter_shape, new_output_shape) = [&] { in PadForTensorCores()167 new_input_shape->dimensions(dnums.input_feature_dimension()); in PadForTensorCores()171 new_input_shape->set_dimensions(dnums.input_feature_dimension(), 4); in PadForTensorCores()177 pad_dim(new_input_shape, dnums.input_feature_dimension()); in PadForTensorCores()
72 Shape new_input_shape = in Run() local75 HloInstruction::CreateTranspose(new_input_shape, input, in Run()
3617 const Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot() local3627 auto new_lhs = add_bitcast(new_input_shape, lhs); in SimplifyConvToDot()