Searched refs:reshaped_activation_shape (Results 1 – 1 of 1) sorted by relevance
406 Shape reshaped_activation_shape = activation->shape(); in HandleConvolution() local407 ShapeUtil::AppendMajorDimension(group_size, &reshaped_activation_shape); in HandleConvolution()409 int64 new_spatial_dim = reshaped_activation_shape.dimensions().size() - 1; in HandleConvolution()411 reshaped_activation_shape.set_dimensions(activation_input_feature_dim, in HandleConvolution()414 HloInstruction::CreateReshape(reshaped_activation_shape, activation)); in HandleConvolution()