Searched refs:new_input_shape (Results 1 – 7 of 7) sorted by relevance
102 Shape new_input_shape = input->shape(); in ExpandInstruction() local104 new_input_shape.DeleteDimension(dim); in ExpandInstruction()162 HloInstruction::CreateReshape(new_input_shape, input)); in ExpandInstruction()
36 Shape new_input_shape = *output_aligned_input_shape; in HandleReshape() local37 HloInstruction* copied_operand = MakeCopyHlo(operand, new_input_shape); in HandleReshape()
6637 Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot() local6639 simplifier_->UpdateLayout(&new_input_shape); in SimplifyConvToDot()6650 auto new_lhs = add_bitcast(new_input_shape, lhs); in SimplifyConvToDot()
207 Shape* new_input_shape; in TryResolvePaddedShapesForTensorCore() local210 std::tie(new_input_shape, new_filter_shape, new_output_shape) = [&] { in TryResolvePaddedShapesForTensorCore()229 new_input_shape->dimensions(dnums.input_feature_dimension()); in TryResolvePaddedShapesForTensorCore()233 new_input_shape->set_dimensions(dnums.input_feature_dimension(), 4); in TryResolvePaddedShapesForTensorCore()239 pad_dim(new_input_shape, dnums.input_feature_dimension()); in TryResolvePaddedShapesForTensorCore()
75 Shape new_input_shape = in Run() local78 HloInstruction::CreateTranspose(new_input_shape, input, in Run()
65 auto new_input_shape = std::get<InputShape>(new_key); in hit_dynamic() local67 cached_input_shape == new_input_shape && in hit_dynamic()
839 new_input_shape = [1] * (len(input.shape) + 1)841 new_input_shape[0] = 2842 input = input.repeat(new_input_shape)844 new_input_shape[1] = 2845 input = input.unsqueeze(1).repeat(new_input_shape)